现象:
在利用YOLOv9源码进行模型训练的时候,在训练完成后,权重文件夹中的图像和txt文件,val_loss一直为0。
原因:
在训练过程中,为计算验证的loss
修改:
在val_dual.py的197行,将原本注释的代码解开即可。
# Loss
if compute_loss:
preds = preds[1]
#train_out = train_out[1]
loss += compute_loss(train_out, targets)[1] # box, obj, cls
else:
preds = preds[0][1]
此时再次训练便会有val loss