search

NVIDIA-AI-IOT / tf_trt_models

TensorFlow models accelerated with NVIDIA TensorRT
BSD 3-Clause "New" or "Revised" License
686 stars 241 forks source link

Provide Category Index #22

Open Davidnet opened 6 years ago

Davidnet commented 6 years ago

Hi Nvidia team! Your work is amazing!

In the detection notebook, is possible to provide the CATEGORY_INDEX, i.d the mapping of id to category string, it seems somehow the tensorrt changes this id.

Thanks!

ghost commented 6 years ago

Hi,

Thanks for the positive feedback!

As for the category index, you can try checkouting out the branch improved_model_support (and re-install this project).

This doesn't have the category string, but does have a class index that is consistent with the original tensorflow object detection API.