bubbliiiing / yolov7-pytorch

这是一个yolov7的库,可以用于训练自己的数据集。
GNU General Public License v3.0
861 stars 150 forks source link

召回率高、精度低 #68

Open XHC1901 opened 1 year ago

XHC1901 commented 1 year ago

您好,学习了您有关“yolov7训练自己数据集”的视频,我现在的状态是:初始权重是yolov7_weights.pth 、没有冻结训练、优化器用adam、解冻阶段的epoch是500、batch_size是2;想请教一下:可以在哪些方面进行调整尝试,使predict的精度提高?我现在的是:38.48% = AP; score_threhold=0.5 : F1=0.19 ; Recall=96.15% ; Precision=10.71% 。精度特别低

bubbliiiing commented 1 year ago

鹅……用sgd试试吧

neuhsm commented 1 year ago

这个不是,召回率和精度和置信度有关的,score_threhold调高一些就好了

bubbliiiing commented 1 year ago

这AP也太低了,所以训练就没训练好应该