Python IMage MIning
Project description
PIMMI : Python IMage MIning
Library allowing visual search in a corpus of images, from Twitter... or elsewhere.
SIFT interest points, clustering, based on OpenCV and Faiss, multithreaded.
Very preliminary stuff for now.
Demo
# Install dependencies
pip install -r requirements.txt
# --- Play with a very small dataset
# Create a default index structure and fill it with the demo dataset
python3 index_dataset.py --action fill --thread 16 --index "IVF1024,Flat" --save_faiss index/small_dataset.ivf1024 --images_dir demo_dataset/small_dataset
# Query the same dataset on this index
python3 query_dataset.py --simple --thread 16 --load_faiss index/small_dataset.ivf1024 --save_mining index/small_dataset.ivf1024.mining --images_mining --images_root demo_dataset/dataset1
# --- Play with the demo dataset 1
python3 index_dataset.py --action fill --thread 16 --index "IVF1024,Flat" --save_faiss index/dataset1.ivf1024 --images_dir demo_dataset/dataset1
python3 query_dataset.py --thread 16 --load_faiss index/dataset1.ivf1024 --save_mining index/dataset1.ivf1024.mining --images_mining --images_root demo_dataset/dataset1
# Post process the mining results in order to visualize them
python3 fuse_query_results.py
python3 generate_cluster_viz.py
Happy hacking !
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
pimmi-0.0.7.tar.gz
(25.5 kB
view hashes)
Built Distribution
pimmi-0.0.7-py3-none-any.whl
(30.0 kB
view hashes)