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

Question

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

Answer 1

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

检查笔记本https://github.com/fera0013/TrafficLightDetection/blob/master/Traffic-Light-Detection.ipynb 中的def filter_boxes 函数，查看如何按类别过滤框。