1. 问题描述
- 在进行 yolov5 模型训练时会卡在
Transferred342/349items fromweights/yolov5s.pt
处很长时间。
2. 原因分析
- 从
ROOT/utils/general.py
的check_amp()
函数中可知卡住不动的原因为:- 本地
ROOT/data/images/bus.jpg
路径中不存在 bus.jpg 图像。 - 本地不存在 bus.jpg 图像后访问 ‘https://ultralytics.com/images/bus.jpg’ 这张图片时网络很慢,使得代码一直在使用
check_online()
函数检查网络,导致训练卡在这里。
- 本地
3. 解决方法
- 禁止使用
check_online()
函数检查网络,修改utils/general.py
中check_amp()
函数即可。# im = f if f.exists() else 'https://ultralytics.com/images/bus.jpg' if check_online() else np.ones((640, 640, 3)) im = f if f.exists() else 'https://ultralytics.com/images/bus.jpg' if False else np.ones((640, 640, 3))
- 修改后的
check_amp()
函数