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

Posted 2023-03-12

【中文标题】如何在张量流对象检测中仅检测特定类别的对象

【英文标题】：How to detect only objects of a specific category in tensorflow object detection

【发布时间】：2018-03-12 22:39:46

【问题描述】：

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

【问题讨论】：

在物体检测代码文件夹中，有一个可视化文件。您需要编辑该文件并仅保留相关的对象检测值。删除其余部分（将它们标记为 0）。

谢谢，我自己发现了。我很快就会在这里上传一个修改过的笔记本，演示解决方案。

【参考方案1】：

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

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

【讨论】：

我正在尝试运行您的笔记本以仅过滤人员类别，但我被困在 检测测试图像中的交通灯部分。我在ilustrate_detection(image,boxes,classes,model.split('\\')[1]) 行中收到IndexError: list index out of range。我也在设置id=1。你能帮助我吗？谢谢！

@Minoru 如果您在 unix 中运行它，则将 '\\" 替换为 "/" - 它会起作用。