报错问题:
In loss_tal.py: pred_distri, pred_scores = torch.cat([xi.view(feats[0].shape[0], self.no, -1) for xi in feats], 2).split(
(self.reg_max * 4, self.nc), 1)
The error is as follows:
AttributeError: 'list' object has no attribute 'view'
解决方法一:
修改yolov9.yaml文件 :将检测头DualDDetect换成Detect
解决方法二:
更换tain.py为train_dual.py 进行训练。