如何仅在对象检测API Tensorflow中检测人

Question

我正在使用tensorflow对象检测API来检测对象。在我的Windows系统中运行正常。我该如何对其进行更改以仅检测提到的对象，例如，我只想检测人类而不是所有对象。

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

Answer 1

我从您的问题中假设，您不是自己对模型进行微调，而是使用了model zoo中经过预训练的模型！？

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