Skip to content

πŸ“Έ A robust image classifier and object detector tailored for photographers. Harnessing the power of Vision Transformer (ViT) and DEtection TRansformer (DETR), this app brings AI-driven insights to your photos. πŸš€

License

Notifications You must be signed in to change notification settings

lakescorp/ImageLabelIA

Repository files navigation

πŸ“Έ Image Classifier for Photographers πŸ“Έ

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.

πŸ“· Application Screenshots:

Home Capture

Process Capture

Tags applied capture

✨ Features:

  • πŸš€ 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.

πŸ›  Setup

Dependencies:

Installation:

  1. Clone the repository:
git clone https://github.com/lakescorp/ImageLabelIA.git
cd ImageLabelIA
  1. Install the required packages:
pip install transformers torch pillow iptcinfo3

πŸš€ Usage:

Run the main script:

python app.py

Once the GUI launches:

  1. πŸ“‚ Click on "Process Folder" to select a directory with images.
  2. βš™οΈ Choose desired options.
  3. πŸ‘β€πŸ—¨ The application will display each image with its classification and detected objects.
  4. πŸ” You can click on each image to view it in full size.

🀝 Contribution:

Pull requests are welcome. For significant changes, please open an issue first to discuss what you'd like to change.

πŸ“œ License:

MIT

About

πŸ“Έ A robust image classifier and object detector tailored for photographers. Harnessing the power of Vision Transformer (ViT) and DEtection TRansformer (DETR), this app brings AI-driven insights to your photos. πŸš€

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Languages