ImgSearch

Search a folder of images based on a base image using CUDA

Steps:

• python ImgSearch.py
• Input base PNG
• Input folder to check

Program will check all images in folder
Finds top-10 images most similar to original image
Auto-opens the most similar image in default image viewer

Program only checks images that have 1/2/4/8x multiple dimensions of the base image

This is by design, they will be skipped for similarity checks with a similarity of 0

Requirements

• Python 3.10+
• NVIDIA GPU
• HuggingFace User Access Token (READ)

pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121
pip install open_clip_torch pillow huggingface_hub
hf auth login

Launch Options

-s or -simple: Will not show list of similarity for each file, will only display Top 10
-n or -noopen: Will not auto-open the image with the most similarity