For Detectnet, is it possible to only shows person box?

Hi, SSD-mobile net has 1000 items. Is it possible for the display to only show person (ClassID ==1) box?
Do I need to modify any code?
Or, is it easier just to do transfer learning training of “person”?



It’s simpler to modify the code.
You can update the parsing function to allow only ClassID==1.


Thx thx. :>

You can also change the jetson-inference/data/networks/SSD-Mobilenet-v2/ssd_coco_labels.txt so that the classes you want to ignore are called void. It will then ignore those classes and you don’t need to edit any code.

Yes, it is working. :> Thx