A2111 _postprocess_trt部分的问题

Soos

我按照老师之前改的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):部分也要更改一下，如下图：

nvadmin 发表于 2021-4-18 09:20
另外，def detect_dir(dir, trt_ssd, conf_th, vis):部分也要更改一下，如下图：

谢谢老师，问题解决了