python - Detectron2检测后如何选择类

Question

我正在创建一个可以使用detectron2（“faster_rcnn_R_50_DC5_3x”）检测的代码，但我只想让他选择“person”类。如何在检测后制作过滤器只是为了选择我想要的类并在打印时忽略其他类

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

标签： python

Answer 1

用你要添加的类的属性设置一些条件，如果不符合条件则不添加