[Yolov5] ModuleNotFoundError : no module named ‘models' 에러

📌 오류내용

병해충 모델을 Colab이 아닌 로컬환경에서 불러와 이미지를 판별하고자
torch.load("./best_model.pt")를 통해 모델을 업로드하였다.
하지만 아래와 같이 model을 찾을 수 없다는 오류가 발생하였다.

📌 원인

model이 있는 py경로가 아닌 다른 경로에서 load 할 때 발생하는 에러이다.

📌 해결방법

sys.path를 설정해 주면 해결할 수 있다.

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

위와 같이 system path에 모델의 경로를 추가해주면 정상적으로 모델을 load할 수 있다!

📌 결론

app.py

import sys
sys.path.insert(0, './model')
tomato_model = torch.hub.load('./yolov5', 'custom', path='./model/result/tomato.pt', source='local
result = tomato_model(train_img, size = img_size)

이렇게 model의 경로를 지정해주고 사용하고자 하는 모델의 경로를 지정해 주면 해결완료!

📌 폴더 구조

• app.py
  yolov5
  model - result - tomato.pt