Semi Automatic Image Annotation Toolbox with RetinaNet as the suggesting algorithm. The toolbox suggests 80 class objects from the MS COCO dataset using a pretrained RetinaNet model.
-
Clone this repository.
-
In the repository, execute
pip install . --user
. Note that due to inconsistencies with howtensorflow
should be installed, this package does not define a dependency ontensorflow
as it will try to install that (which at least on Arch Linux results in an incorrect installation). Please make suretensorflow
is installed as per your systems requirements. Also, make sure Keras 2.1.3 or higher is installed. -
Download the pretrained weights and save it in /snapshots.
-
Tensorflow >= 1.7.0
-
OpenCV >= 3.4.0
-
Keras >= 2.1.3
For, Python >= 3.5
python main.py