Skip to content

Latest commit

 

History

History
167 lines (131 loc) · 6.92 KB

README.md

File metadata and controls

167 lines (131 loc) · 6.92 KB
Raw

DigitalSreeni Image Annotator and Toolkit

Python Version License PyPI version

A powerful and user-friendly tool for annotating images with polygons and rectangles, built with PyQt5. Now with additional supporting tools for comprehensive image processing and dataset management.

DigitalSreeni Image Annotator Demo

Watch the demo: Watch the demo video

@DigitalSreeni Dr. Sreenivas Bhattiprolu

Features

  • Semi-automated annotations with SAM-2 assistance (Segment Anything Model) — Because who doesn't love a helpful AI sidekick?
  • Manual annotations with polygons and rectangles — For when you want to show SAM-2 who's really in charge.
  • Save and load projects for continued work.
  • Import existing COCO JSON annotations with images.
  • Export annotations to various formats (COCO JSON, YOLO v8, Labeled images, Semantic labels, Pascal VOC).
  • Handle multi-dimensional images (TIFF stacks and CZI files).
  • Zoom and pan for detailed annotations.
  • Support for multiple classes with customizable colors.
  • User-friendly interface with intuitive controls.
  • Change the application font size on the fly — Make your annotations as big or small as your caffeine level requires.
  • Dark mode for those late-night annotation marathons — Who needs sleep when you have dark mode?
  • Pick appropriate pre-trained SAM2 model for flexible and improved semi-automated annotations.
  • Additional supporting tools:
    • Annotation statistics for current annotations
    • COCO JSON combiner
    • Dataset splitter
    • Stack to slices converter
    • Image patcher
    • Image augmenter

Installation

You can install the DigitalSreeni Image Annotator directly from PyPI:

pip install digitalsreeni-image-annotator

The application uses the Ultralytics library, so there's no need to separately install SAM2 or PyTorch, or download SAM2 models manually.

Usage

  1. Run the DigitalSreeni Image Annotator application:

    digitalsreeni-image-annotator

    or

    sreeni

    or

    python -m digitalsreeni_image_annotator.main

  2. Using the application:

    • Click "New Project" or use Ctrl+N to start a new project.
    • Use "Add New Images" to import images, including TIFF stacks and CZI files.
    • Add classes using the "Add Classes" button.
    • Select a class and use the Polygon or Rectangle tool to create manual annotations.
    • To use SAM2-assisted annotation:
      • Select a model from the "Pick a SAM Model" dropdown. It's recommended to use smaller models like SAM2 tiny or SAM2 small. SAM2 large is not recommended as it may crash the application on systems with limited resources.
      • Note: When you select a model for the first time, the application needs to download it. This process may take a few seconds to a minute, depending on your internet connection speed. Subsequent uses of the same model will be faster as it will already be cached locally, in your working directory.
      • Click the "SAM-Assisted" button to activate the tool.
      • Draw a rectangle around objects of interest to allow SAM2 to automatically detect objects.
      • Note that SAM2 provides various outputs with different scores, and only the top-scoring region will be displayed. If the desired result isn't achieved on the first try, draw again.
      • For low-quality images where SAM2 may not auto-detect objects, manual tools may be necessary.
    • Edit existing annotations by double-clicking on them.
    • Save your project using "Save Project" or Ctrl+S.
    • Use "Open Project" or Ctrl+O to load a previously saved project.
    • Click "Import Annotations with Images" to load existing COCO JSON annotations along with their images.
    • Use "Export Annotations" to save annotations in various formats (COCO JSON, YOLO v8, Labeled images, Semantic labels, Pascal VOC).
    • Access additional tools under the Tools menu bar:
      • Annotation Statistics
      • COCO JSON Combiner
      • Dataset Splitter
      • Stack to Slices Converter
      • Image Patcher
      • Image Augmenter
    • Each tool opens a separate UI to guide you through the respective task.
    • Access the help documentation by clicking the "Help" button or pressing F1.
    • Explore the interface – you might stumble upon some hidden gems and secret features!

  3. Keyboard shortcuts:

    • Ctrl + N: Create a new project
    • Ctrl + O: Open an existing project
    • Ctrl + S: Save the current project
    • Ctrl + W: Close the current project
    • Ctrl + Shift + S: Open Annotation Statistics
    • F1: Open the help window
    • Ctrl + Wheel: Zoom in/out
    • Hold Ctrl and drag: Pan the image
    • Esc: Cancel current annotation, exit edit mode, or exit SAM-assisted annotation
    • Enter: Finish current annotation, exit edit mode, or accept SAM-generated mask
    • Up/Down Arrow Keys: Navigate through slices in multi-dimensional images

Development

For development purposes, you can clone the repository and install it in editable mode:

  1. Clone the repository:

    git clone https://github.com/bnsreenu/digitalsreeni-image-annotator.git
cd digitalsreeni-image-annotator

  2. Create a virtual environment (optional but recommended):

    python -m venv venv
source venv/bin/activate  # On Windows, use `venv\Scripts\activate`

  3. Install the package and its dependencies in editable mode:

    pip install -e .

Contributing

Contributions are welcome! Please feel free to submit a Pull Request.

  1. Fork the repository
  2. Create your feature branch (git checkout -b feature/AmazingFeature)
  3. Commit your changes (git commit -m 'Add some AmazingFeature')
  4. Push to the branch (git push origin feature/AmazingFeature)
  5. Open a Pull Request

License

This project is licensed under the MIT License - see the LICENSE file for details.

Acknowledgments

  • Thanks to all my YouTube subscribers who inspired me to work on this project
  • Inspired by the need for efficient image annotation in computer vision tasks

Contact

Dr. Sreenivas Bhattiprolu - @DigitalSreeni

Project Link: https://github.com/bnsreenu/digitalsreeni-image-annotator

Citing

If you use this software in your research, please cite it as follows:

Bhattiprolu, S. (2024). DigitalSreeni Image Annotator [Computer software]. https://github.com/bnsreenu/digitalsreeni-image-annotator

@software{digitalsreeni_image_annotator,
  author = {Bhattiprolu, Sreenivas},
  title = {DigitalSreeni Image Annotator},
  year = {2024},
  url = {https://github.com/bnsreenu/digitalsreeni-image-annotator}
}