问题遇到的现象和发生背景
yolop 可以识别车辆,我尝试应用BDD100K数据集识别多目标,比如车辆、行人、红绿灯,但是识别出来的类型全部为0(car)
问题相关代码,请勿粘贴截图
在yolop代码中,
将self.nc = 1 改为 self.nc = 13;
将single_cls = True# just detect vehicle 改为 single_cls = False
运行结果及报错内容
训练结束后,行人、红绿灯可以正确框出来,但是检测出的类型cls都为0(car)
我的解答思路和尝试过的方法
请问yolop是否支持多目标识别,还是只能识别车辆;目前遇到的问题是想将检测出来的目标分类正确,而不全是0;请问应该怎么调整代码,实现多目标检测。
我想要达到的结果
另外,model.nc = 1这个代表什么,是否需要调整?
注:
id_dict = {'person': 0, 'rider': 1, 'car': 2, 'bus': 3, 'truck': 4, 'bike': 5, 'motor': 6, 'tl_green': 7, 'tl_red': 8,
'tl_yellow': 9, 'tl_none': 10, 'traffic sign': 11, 'train': 12}
id_dict_single = {'car': 0, 'bus': 1, 'truck': 2,'train': 3}