The object detection notebook demonstrates, how models pretrained on the COCO dataset can be used to detect objects on test images. However, the models in the notebook return boxes for detected objects of all categories in the COCO set. How can I use the code to return boxes for objects of only one category? I.e. How can I get boxes for objects of which the model is sure that they are e.g. persons?
1 Answers
I have just implemented the solution myself.
check the def filter_boxes function in the notebook https://github.com/fera0013/TrafficLightDetection/blob/master/Traffic-Light-Detection.ipynb, to see how the boxes can be filtered by category.
Related