如何在物体检测中只检测人类 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"
.
我正在使用 tensorflow
对象检测 API 来检测对象。它在我的 windows 系统中运行良好。我如何更改它以仅检测提到的对象,例如,我只想检测人而不是所有对象。
根据此 answer 中的第一条评论,我检查了可视化文件,但没有找到与对象类别相关的任何内容。然后我查看了 category_util.py
并发现有 csv
文件从中加载所有类别但没有在项目中找到这个 csv
文件。谁能指出我正确的方向。谢谢
根据你的问题,我假设你没有自己微调你的模型,而只是使用了 model zoo!?
中的预训练模型在这种情况下,我认为模型已经检测到人类和其他物体,而您希望这些其他物体消失!?为此,您只需删除所有不需要的 类 来更改 label_map.pbtxt
。如果您不确定在哪里可以找到此文件,请查看您的 .config
文件并搜索 label_map_path="PATH"
.