克隆YOLOv5官方仓库训练时遇ModuleNotFoundError报错求助

核心原因 #

Python无法找到models模块，通常是工作目录错误或文件结构不完整导致的。

具体解决步骤 #

• 切换到YOLOv5根目录执行命令
  克隆的YOLOv5仓库有完整的目录结构，必须确保当前终端的工作目录是仓库根目录（包含train.py、models/、data/的文件夹）。

  • Windows命令行：

    cd C:\Users\nived\Untitled Folder 1\yolov5
    python train.py --img 416 --batch 16 --epochs 150 --data relative/path/to/data.yaml --weights yolov5s.pt --cache
    

  • Jupyter环境：先执行目录切换命令，再运行训练脚本：

    %cd C:\Users\nived\Untitled Folder 1\yolov5
    !python train.py --img 416 --batch 16 --epochs 150 --data relative/path/to/data.yaml --weights yolov5s.pt --cache
    

• 验证文件结构完整性
  检查train.py所在的文件夹下是否存在models/目录，里面应该包含yolo.py、common.py等文件。如果缺失，说明克隆不完整，重新克隆仓库并安装依赖：

  git clone https://github.com/ultralytics/yolov5.git
  cd yolov5
  pip install -r requirements.txt
  

• 手动添加Python路径（备用方案）
  如果无法切换目录，可在train.py开头添加以下代码，强制Python识别当前目录下的模块：

  import sys
  sys.path.insert(0, './')
  

  或者在运行命令时临时设置Python路径（Windows）：

  set PYTHONPATH=C:\Users\nived\Untitled Folder 1\yolov5 && python train.py --img 416 --batch 16 --epochs 150 --data relative/path/to/data.yaml --weights yolov5s.pt --cache
  

内容的提问来源于stack exchange，提问作者NIVEDITHA J