raccoon_dataset icon indicating copy to clipboard operation
raccoon_dataset copied to clipboard

Read classes from label map

Open JR-378 opened this issue 6 years ago • 2 comments

I grew tired of doing copy and paste from the label map to the script.

There is additional command line parameter to provide the path to the label map. The label map will be read once in the script, parsed into a dictionary and passed to the create_tf_example method.

The names are the keys in the dictionary and the IDs are the values, which makes it work much the same way as the previous method.

All tests passed and tfrecords have been created without issues.

Edit: I recently started training another model and everything seems to be working.

JR-378 avatar Jun 16 '19 16:06 JR-378

you could add: import tensorflow.compat.v1 as tf to have tf.app working, if this is ever merged

jarvis-owl avatar Dec 24 '19 09:12 jarvis-owl

Probably never will be. Anyways, my goal was to just share the code, as I am using it in my private repository quite often and without failure.

JR-378 avatar Jan 29 '20 03:01 JR-378