https://github.com/tomasrasymas/simimg

Similar image search
https://github.com/tomasrasymas/simimg

image-processing python similarity-search

Last synced: about 2 months ago
JSON representation

Similar image search

• Host: GitHub
• URL: https://github.com/tomasrasymas/simimg
• Owner: tomasrasymas
• Created: 2019-06-26T19:45:14.000Z (over 5 years ago)
• Default Branch: master
• Last Pushed: 2022-12-08T05:17:25.000Z (about 2 years ago)
• Last Synced: 2024-08-01T13:25:35.529Z (5 months ago)
• Topics: image-processing, python, similarity-search
• Language: Python
• Size: 5.97 MB
• Stars: 48
• Watchers: 2
• Forks: 7
• Open Issues: 9
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# SimImg (Similar Image)

SimImg is a set of tools used to find similar images. In this repo I am providing example using DeepFashion dataset. If you have image with some kind of outfit in it, you can search for similar ones by visual similarity.

## Pipeline

* Every image in image dataset is converted to multidimensional feature vector (`pre-trained ResNet50`)

* Source image is converted to same feature vector

* Similarity between source image and every dataset images is calculated (`cosine similarity`)

* Most similar images are selected as result

## Files description

* `config.py` - main config file of a system

* `download_dataset.py` - downloads DeepFashion dataset into local hard drive (`data/deepfashion/`)

* `generate_features_db.py` - generates feature vectors for every image in a dataset (`data/features_db.pckl`)

* `search.py` - loads generated features DB and makes search for similar images in DB

* `image_2_vector.py` - converts image to its feature vector

* `generate_td_projector_data.py` - generates tensorboard projector files two view feature vectors in thier embeddings space (`data/projector/`) 

 

## Image embeddings

Output of pre-trained model: `ResNet50(weights='imagenet', include_top=False, pooling='avg')`

## TensorBoard projector

It is very resource intensive task. Be careful when selecting number of images to add to projector, cause in order to use TB projector all images are merged into singe montage image. 

![montage](docs/montage.png)

```

python generate_tb_projector_data.py --path data/montage --output data/projector --dataset_num 3000

cd data/projector

tensorboard --logdir .

```

![tensorboard](docs/tensorboard.png)

## Google Colab

To speed up features DB generation Google Colab with it's free GPU was used. Notebook is located in colab_notebooks folder. After features DB is created copy it to `data/` folder. 

## Search for similar images (interactive)

```

python search.py --num 20

```

Search result is similar images merged into single montage image, keeping source image the first one.

### Sample 1

![source_1](docs/sample_1_source.jpg)

![results_1](docs/sample_1_results.png)

### Sample 2

![source_2](docs/sample_2_source.jpg)

![results_2](docs/sample_2_results.png)

### Sample 3

![source_3](docs/sample_3_source.jpg)

![results_3](docs/sample_3_results.png)

## Dataset

https://drive.google.com/open?id=0B7EVK8r0v71pQ2FuZ0k0QnhBQnc

http://mmlab.ie.cuhk.edu.hk/projects/DeepFashion.html