yolov5 :我能在已经做过定制培训的模型上增加更多的划时代吗？

Question

我试着用3万张图像制作图像检测模型。我刚试了5次，训练了大约5个小时。

问题是，如果我想简单地在同一个模型上增加更多的时代，我怎么能做到呢？有什么方法我可以加载自定义的训练模型吗？

因为这需要很长时间，我希望我能继续训练从5个时代已经完成的训练。我正在使用colab，在训练过程中，我经历了几次失去联系的经历，所以我害怕从训练开始就给予更多的划时代。

Answer 1

是的，您可以使用标准的yolo训练命令，并在-权重参数中为您以前的训练模型提供路径。

!python train.py --img 960 --batch 20 --epochs 10 --data dataset.yaml --weights your_model.pt --cache --exist-ok