search

DocF / multispectral-object-detection

Multispectral Object Detection with Yolov5 and Transformer
GNU Affero General Public License v3.0
315 stars 60 forks source link

关于训练文件缺失的问题 #5

Closed 116508 closed 2 years ago

116508 commented 2 years ago

Traceback (most recent call last): File "/home/fhl/src/rgbt-det/multispectral-object-detection/train.py", line 1028, in train_rgb_ir(hyp, opt, device, tb_writer) File "/home/fhl/src/rgbt-det/multispectral-object-detection/train.py", line 519, in train_rgb_ir model = Model(opt.cfg or ckpt['model'].yaml, ch=6, nc=nc, anchors=hyp.get('anchors')).to(device) # create File "/home/fhl/src/rgbt-det/multispectral-object-detection/models/yolo_test.py", line 188, in init self.model, self.save = parse_model(deepcopy(self.yaml), ch=[ch]) # model, savelist File "/home/fhl/src/rgbt-det/multispectral-object-detection/models/yolo_test.py", line 529, in parse_model elif m is CMAFF: NameError: name 'CMAFF' is not defined

在common.py文件中发现CMAFF未定义,请问是什么原因导致的?

DocF commented 2 years ago

谢谢你的提醒，CMAFF其实是我另一篇论文的模型，这里不小心放进来了。 过段时间我会清理一遍代码，以确保此类问题不再发生。

116508 commented 2 years ago

非常感谢你的回复，你的开源工作将给我带来巨大的帮助，谢谢

wlc-git commented 2 years ago

How do you do? May I ask why CLS is always 0 when I train

DocF commented 2 years ago

非常感谢你的回复，你的开源工作将给我带来巨大的帮助，谢谢

已经清理一遍代码并提交，你可以使用当前版本的代码进行训练、测试。