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

Question

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

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

Answer 1

我从你的问题中假设你没有自己微调你的模型，而是使用了来自 model zoo 的预训练模型。 !？

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