jveitchmichaelis / deeplabel

A cross-platform desktop image annotation tool for machine learning
205 stars 39 forks source link

Allow users to set class name-ID mapping during export #9

Open jveitchmichaelis opened 4 years ago

jveitchmichaelis commented 4 years ago

This affects the COCO and Pascal VOC exporters. Currently the output class ID is dependent on the order in which they were created in deeplabel. It may be preferable to, in some cases, force this to something else, for example provide a pbtxt file like https://github.com/tensorflow/models/blob/master/research/object_detection/data/mscoco_label_map.pbtxt

The darknet exporter does not have this problem as it uses a .names file to control this. GCP is also not affected as it uses class names directly.

Should probably allow minimally:

and infer based on extension.