如何在张量流对象检测中仅检测特定类别的对象

Question

object detection notebook演示了models上预先训练的COCO dataset如何用于检测测试图像上的对象。但是，笔记本中的模型会返回COCO集中所有类别的检测对象的框。如何使用代码返回仅一个类别的对象的框？即我如何获得模型确定它们是例如对象的框。人？

Answer 1

我自己刚刚实施了解决方案。

检查笔记本https://github.com/fera0013/TrafficLightDetection/blob/master/Traffic-Light-Detection.ipynb中的def filter_boxes功能，了解如何按类别过滤这些框。