【YOLO】拾遗(五)

0 YOLO系列笔记

1 前言

  本文汇总在使用yolo过程中遇到的一些问题,时时更新。

2 训练中断怎么办?

  如果自己的电脑性能不够,导致训练过程中蓝屏了;或者是因为什么必要的原因必须关掉电脑,那么就会面临训练了很久的YOLO模型被迫中断,好在YOLOv5中已经集成了断点续传的功能,遇到这种情况,完全不用担心。
  解决办法就是稍微改动一下train.py的源码,将--resume参数的默认值改为True,然后再运行之前训练数据集的指令即可。它会直接检测上一次的遗留,然后接着断点继续训练。
在这里插入图片描述

注意:如果直接在命令行参数传入--resume参数的值,经过测试,发现会报错,可能是什么代码没有写好。
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/ZHOU_YONG915/article/details/132004921
今日推荐