Ssd-mobilenet-v2 object detection

I am trying to run ssd mobilenetv2 model on jetson nano for object detection. I needed help with a couple of basic queries.

  1. How many labels is the model trained on? Is there a label.txt file with the list of labels
  2. What dataset is this model (the one that is pre-installed on jetson nano) trained on? Right now, I am getting inconsistent object detections. Is there a way to improve the accuracy of the model?


Hi @tauseefahmad12, if you mean the one from jetson-inference, it is trained on the 90-class MS COCO dataset. The labels.txt file can be found here:

To improve the detections, you could increase the confidence threshold with the --threshold parameter, or collect data and train your own model with