A2111 _postprocess_trt部分的问题

我按照老师之前改的cls=cls-1后,得出的mAP很低,而且只有road_sign不为0,明显之前的结果,而且我在ssd-class里改变顺序结果也不改变,而且也不能识别图片上的汽车了。
如果是cls没有后面没有-1的话,能识别图片的汽车并画框,而且结果里detection-result的汽车识别度也很高,就是road-sign没有名字
请老师能不能再详细的解释一下为什么会出现这种情况和解决方案。

这个问题的本质是模型输出是5类了,可能是添加了一个background类别,你在label列表里面添加一个background类别试一下,在utils/ssd_classes.py 里面改,然后把cl = cl -1 去掉,试一下

比如我的改成了这样:

另外,def detect_dir(dir, trt_ssd, conf_th, vis):部分也要更改一下,如下图:

谢谢老师,问题解决了:lol