【问题记录】Yolov5 训练时卡在 Transferred 342/349 items from weights/yolov5s.pt 处很长时间

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() 函数