如何在物体检测中只检测人类 API Tensorflow

How to only detect humans in object detection API Tensorflow

我正在使用 tensorflow 对象检测 API 来检测对象。它在我的 windows 系统中运行良好。我如何更改它以仅检测提到的对象,例如,我只想检测人而不是所有对象。

根据此 answer 中的第一条评论,我检查了可视化文件,但没有找到与对象类别相关的任何内容。然后我查看了 category_util.py 并发现有 csv 文件从中加载所有类别但没有在项目中找到这个 csv 文件。谁能指出我正确的方向。谢谢

根据你的问题,我假设你没有自己微调你的模型,而只是使用了 model zoo!?

中的预训练模型

在这种情况下,我认为模型已经检测到人类和其他物体,而您希望这些其他物体消失!?为此,您只需删除所有不需要的 类 来更改 label_map.pbtxt。如果您不确定在哪里可以找到此文件,请查看您的 .config 文件并搜索 label_map_path="PATH".