This repository is designed for photographers π·! It provides a graphical application to classify and detect objects in images. Use the power of the Vision Transformer (ViT)
for classification and the DEtection TRansformer (DETR)
for object detection. Easily select a folder with your images and let the application highlight the magic within them.
- π Multi-threaded image processing.
- π Progress bar and process status updates.
- π€ Option to trust AI and apply all suggestions.
- πΌ Thumbnails of images on the canvas.
- π Ability to stop the ongoing processing.
- π Dark mode.
- Clone the repository:
git clone https://github.com/lakescorp/ImageLabelIA.git
cd ImageLabelIA
- Install the required packages:
pip install transformers torch pillow iptcinfo3
Run the main script:
python app.py
Once the GUI launches:
- π Click on "Process Folder" to select a directory with images.
- βοΈ Choose desired options.
- πβπ¨ The application will display each image with its classification and detected objects.
- π You can click on each image to view it in full size.
Pull requests are welcome. For significant changes, please open an issue first to discuss what you'd like to change.