问题
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
回答1:
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"
.
来源:https://stackoverflow.com/questions/53281990/how-to-only-detect-humans-in-object-detection-api-tensorflow