如何用之前实验中发现的权重训练YOLOv5模型



我正在训练YOLOv5模型。我把它训练了50个时代。但我认为mAP仍然可以增加。我如何使用从50个历元实验中获得的权重(best.pt(来运行60个历元的新实验?我正在使用此代码。如何设置权重?

!python train.py --batch 27 --epochs 60  --data /content/yolov5/data.yaml --weights yolov5x.pt 

50 epoch best.pt文件是在YOLOv5l上运行模型获得的。现在我在YOLOv5x上运行它。它有用吗?

您可以将权重参数替换为最佳.pt权重的位置!python train.py--批处理27--历元60--data/content/yolov5/data.yaml--权重//最佳位置.pt/

相关内容

  • 没有找到相关文章

最新更新