在Detectron2上进行检测后如何选择类别

时间：2020-10-20 17:20:25

标签： python

我创建了一个可以使用来自detectron2的检测代码（“ faster_rcnn_R_50_DC5_3x”），但是我只希望他选择“人”类。检测到后如何仅选择我想要的类而忽略打印中的其他类

enter image description here

例如，在该图像上它检测到了球和汽车，而我只想打印“人”检测结果

1 个答案:

答案 0 :(得分：0)

使用您要添加的类的属性设置一些条件，如果它们不符合条件，则不会添加它们

相关问题
最新问题