ModuleNotFoundError : No module named 'models'

Question

I'm using YOLOv5, clone from its official github repo.

During the training I got the following error:

Traceback (most recent call last):
  File "C:\Users\nived\Untitled Folder 1\train.py", line 40, in <module>
    import models
ModuleNotFoundError: No module named 'models'

Train command:

!python train.py --img 416 --batch 16 --epochs 150 --data relative/path/to/data.yaml --weights yolov5s.pt --cache*

Answer 1

Please add mentioned code in the top two lines of the detect.py file and it will work then,

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

For more details, you can visit link from YOLOv5 Repository