本文介绍了如何修改YOLOV5的train.py文件，通过在训练的前n个epoch不计算mean Average Precision (mAP)，以此来加快训练速度。以4个epoch为例，前4次迭代不计算mAP，从第5个epoch开始计算，显著减少了每次迭代的时间消耗。
将train.py文件中的：
if RANK in {-1, 0}:
# mAP
callbacks.run('on_train_epoch_end', epoch=epoch)
ema.update_attr(model, include=['yaml', 'nc', 'hyp', 'names', 'stride', 'class_weights'])
final_epoch = (epoch + 1 == epochs) or stopper.possible_stop
if not noval or final_epoch: # Calculate mAP修改为：
if RANK in {-1, 0}:
# mAP
callbacks.run('on_train_epoch_end', epoch=epoch)
ema.update_attr(model, include=['yaml', 'nc', 'hyp', 'names', 'stride', 'class_weights'])
final_epoch = (epoch + 1 == epochs) or stopper.possible_stop
if (not noval and epoch >= 4) or final_epoch: # Calculate mAP</
最低0.47元/天 解锁文章
4594
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
