前言
在slowfast中,是先调用faster rcnn,框出人的位置,再把这些人丢进行slowfast的网络,但是再slowfast调用faster rcnn中,官方默认的阈值为0.9,也就是说,检测为人的概率为90%,才能丢进slowfast。这样就导致当画面中人比较多,比较小的时候,只有少数人能被框出来,那么调整阈值,以让更多的人被检测出来。
代码修改
在目录:SlowFast-master/slowfast/visualization/predictor.py:
修改如下:
#self.cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = cfg.DEMO.DETECTRON2_THRESH
self.cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.7
检测结果对比
下面是官方默认效果
下面是修改阈值后的结果