https://github.com/AlvaroCavalcante/auto_annotate
Labeling is boring. Use this tool to speed up your next object detection project!
https://github.com/AlvaroCavalcante/auto_annotate
annotation-tool computer-vision dataset-generation labeling-tool object-detection pattern-recognition python python3 semi-supervised-learning tensorflow weak-supervision
Last synced: about 1 month ago
JSON representation
Labeling is boring. Use this tool to speed up your next object detection project!
- Host: GitHub
- URL: https://github.com/AlvaroCavalcante/auto_annotate
- Owner: AlvaroCavalcante
- License: apache-2.0
- Created: 2019-12-23T20:53:15.000Z (over 6 years ago)
- Default Branch: master
- Last Pushed: 2024-07-05T18:54:21.000Z (almost 2 years ago)
- Last Synced: 2025-09-24T21:55:09.629Z (7 months ago)
- Topics: annotation-tool, computer-vision, dataset-generation, labeling-tool, object-detection, pattern-recognition, python, python3, semi-supervised-learning, tensorflow, weak-supervision
- Language: Jupyter Notebook
- Homepage: https://medium.com/p/acf410a600b8#9e0e-aaa30a9f4b7a
- Size: 54.2 MB
- Stars: 162
- Watchers: 4
- Forks: 34
- Open Issues: 7
-
Metadata Files:
- Readme: README.md
- License: LICENSE.txt
Awesome Lists containing this project
- awesome-auto-annotation - AlvaroCavalcante/auto_annotate
README
[](https://colab.research.google.com/github/AlvaroCavalcante/auto_annotate/blob/master/assets/auto_annotate_example.ipynb)
# Auto Annotation Tool for TensorFlow Object Detection
Are you tired to label your images by hand when working with object detection? Have hundreds or thousands of images to label? Then this project will make your life easier, just create some annotations and let the machine do the rest for you!
# Contents
- [How it works](#how)
- [Prerequisites](#prerequisites)
- [Installation](#installation)
- [Usage](#usage)
- [Command line](#command-line)
- [Code](#code)
- [Google Colab](#colab)
- [Contribute](#contribute)
This auto annotation tool is based on the idea of a semi-supervised architecture, where a model trained with a small amount of labeled data is used to produce the new labels for the rest of the dataset.
As simple as that, the library uses an initial and simplified object detection model to generate the XML files with the image annotations (considering the PASCAL VOC format).
Besides that, it's possible to define a confidence threshold for the detector, acting as a trade-off for the generated predictions.
If you want to know more technical details about the project, please, refer to my [Medium article](https://medium.com/p/acf410a600b8#9e0e-aaa30a9f4b7a).
To use this library you will need a pre-trained object detection model with a subsample of your dataset. As a semi-supervised solution, it's impossible to avoid manual annotation, but you'll need to label just a small amount of your data.
It's hard to determine the number of images to label manually, once it depends on the complexity of your problem. If you want to detect dogs and cats and have 2000 images in your dataset, for example, probably 200 images are enough (100 per class). On the other hand, if you have dozens of classes or objects that are hard to detect, you should need more manual annotations to see the benefits of the semi-supervised approach.
After training this initial model, export your best checkpoint to the [SavedModel](https://www.tensorflow.org/guide/saved_model) format and you'll be ready to use the auto annotation tool!
It's recommended to use a Python [virtual environment](https://docs.python.org/3/library/venv.html) to avoid any compatibility issue with your TensorFlow version.
In your environment, you can install the project using pip:
```bash
pip install auto-annotate
```
You can use this tool either from the command line or directly in your Python code. For both, you'll have the same set of parameters:
- saved_model_path: The path of the **saved_model** folder with the initial model.
- label_map_path: The path of the **label_map.pbtxt** file.
- imgs_path: The path of the folder with your dataset images to label.
- xml_path (**optional**): Path to save the resulting XML files. The default behavior is to save in the same folder of the dataset images.
- threshold: Confidence threshold to accept the detections made by the model. the defaults is 0.5.
To use this tool from the command line, you just need to run:
```bash
python -m auto_annotate --label_map_path /example/label_map.pbtxt \
--saved_model_path /example/saved_model \
--imgs_path /example/dataset_images \
--xml_path /example/dataset_labels \
--threshold 0.65
```
To use this tool from your Python code, check the following code snippet:
```python
from auto_annotate import AutoAnnotate
ann_tool = AutoAnnotate(
saved_model_path = '/example/saved_model',
label_map_path = '/example/label_map.pbtxt',
images_path = '/example/dataset_images',
xml_path = '/example/dataset_labels',
detection_threshold = 0.65)
ann_tool.generate_annotations()
```
For a complete working example, you can refer to this [Google Colab Notebook](https://colab.research.google.com/drive/14qgA9IUYCVAALJmJabvQ9sDxrKxEezwP?usp=sharing), where I share the details about installlation and setup.
Contributions are welcome! Feel free to open a new issue if you have any problem to use the library of find a bug!
You can also use the [discussions](https://github.com/AlvaroCavalcante/auto_annotate/discussions) section to suggest improvements and ask questions! If you find this library useful, don't forget to give it a :star: or support the project:
