我正在使用Tensorflow API来检测对象,但是只希望检测框中的人。
我尝试了之前提出的解决方案,例如::How to only detect humans in object detection API Tensorflow
我放弃了除人以外的所有其他类,但是它对我不起作用,并且我将num_class更改为1,它也没有作用。
当我将num_class更改为1时,它将返回给我名为NA的框。
更新:我使用的是经过预先训练的模型,名为“ ssd_mobilenet_v1_coco”。
答案 0 :(得分:0)
我再次搜索堆栈溢出后发现了一些问题,遇到了How can i use tensorflow object detection to only detect persons?,它清楚地解决了我的问题。