找回密码
 立即注册
发表于 2021-4-17 23:55:05
8824
本帖最后由 Soos 于 2021-4-17 23:59 编辑

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



Screenshot 2021-04-17 at 11.38.46 PM.png
Screenshot 2021-04-17 at 11.42.50 PM.png
Screenshot 2021-04-17 at 11.38.28 PM.png
Screenshot 2021-04-17 at 11.43.10 PM.png
使用道具 举报 回复
发表于 2021-4-18 08:18:29
这个问题的本质是模型输出是5类了,可能是添加了一个background类别,你在label列表里面添加一个background类别试一下,在utils/ssd_classes.py 里面改,然后把cl = cl -1 去掉,试一下
使用道具 举报 回复 支持 反对
发表于 2021-4-18 09:18:41
比如我的改成了这样:
微信截图_20210418091821.png
使用道具 举报 回复 支持 反对
发表于 2021-4-18 09:20:28
另外,def detect_dir(dir, trt_ssd, conf_th, vis):部分也要更改一下,如下图:
微信图片_20210418091949.png
使用道具 举报 回复 支持 反对
发表于 2021-4-18 10:26:33
nvadmin 发表于 2021-4-18 09:20
另外,def detect_dir(dir, trt_ssd, conf_th, vis):部分也要更改一下,如下图:

谢谢老师,问题解决了
使用道具 举报 回复 支持 反对
发新帖
您需要登录后才可以回帖 登录 | 立即注册