How to only detect humans in object detection API Tensorflow

user_6243129 0
Viewed 2742

I am using tensorflow object detection API to detect objects. It is working fine in my windows system. How can I make changes in it to only detect mentioned objects, for example, I only want to detect humans and not all the objects.

As per the 1st comment in this answer, I have checked the visualization file but didn't find anything related to categories of objects. Then I looked into category_util.py and found out that there is csv file from which all the categories are being loaded but didnt found this csv file in the project. Can anyone please point me into the right direction. Thanks

object-detection-api tensorflow
1 Answers

I assume from your question, that you did not finetune your model yourself, but just used a pretrained one from the model zoo!?

In this case, I think the model already detect humans AND other objects and you want these other objects to disappear!? For doing so, you just have to change your label_map.pbtxt by deleting all classes which you don't need. If you are not sure where to find this file have a look into your .config file and search for label_map_path="PATH".

user_102401010
Related