Welcome to the auto-annotate images for object detection!

Requirements

• You will need to clone the tensorflow repository: https://github.com/tensorflow/models
• Install the dependencies: https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/installation.md
• Install the xElementTree dependency via pip

How to run

• Copy and paste the file generate_xml.py and visualization_utils.py into the research/object_detection/utils in the tensorflow repo.
• Add your pre-treined model and label map into the 'graphs' folder.
• Add the images you want to label into the images folder
• Change the xml path in generate_xml.py to put your own local path.
• Inside the auto_annotate folder run: python3 scripts/detection_images.py
• If everything is ok you will see the inference results and the xml in your respective folders!
• If you have trouble or doubt see my tutorial on medium: https://medium.com/@alvaroleandrocavalcante/auto-annotate-images-for-tensorflow-object-detection-19b59f31c4d9?sk=0a189a8af4874462c1977c6f6738d759