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