PicFinder

A simple windows application to search for images in a directory.

Features

• Search for text in images using OCR. Tested with English, Traditional Chinese, Simplified Chinese.
• Search for objects in images using YOLO11. Labels from COCO.
• Search for images using its class. Labels from ImageNet.
• Supported image formats: formats supported by OpenCV: bmp, dib, jpeg, jpg, jpe, jp2, png, webp, avif, pbm, pgm, ppm, pxm, pnm, pfm, sr, ras, tiff, tif, exr, hdr, pic.

Usage

1. Download the latest release
2. Run the application
3. Select the directory to Index using models (only required once)
4. Search

If you clone the repository:

1. Install the required packages using Poetry. For cpu version, use poetry install --with cpu,dev. For gpu version, use poetry install --with gpu,dev.
2. Put the ONNX format YOLO11 models in the models directory. This can be done using the download_models.py script.

Note

The first time you run the application, it will take some time to index the images in the directory.

Only YOLO11n and YOLO11n COCO models are included in the minimal release. For more models, download the ONNX format models and put them in the models directory.

EXE creation

1. Run build.py to create the exe file. The exe file will be in the main.dist directory.
2. Run build.iss to create the installer. The installer will be in the installer_dist directory.

Details

• OCR is done using RapidOCR
• Object detection adapted from ONNX-YOLOv8-Object-Detection
• Search tokenizer from Simple
• Object detection and image classification model from YOLO11