Colab으로 모델을 학습시킨 후 best_model.pt 파일을 이용해 병해충을 판별해주고 Flask서버를 통해서 결과를 넘겨주기로 하였다.
Colab에서 .pt 모델만을 다운 받아 model/best_model.pt에 위치시킨 후 load를 해보았다.
app.py에서 위의 코드처럼 모델을 가져오고자 하였다.
하지만...
ModuleNotFoundError: No module named 'models'
models 모듈을 찾을 수 없다는 에러가 계속 발생하였다.
best_model.pt파일 + yolov5 파일
애초에 .pt파일을 실행시키려면
Yolov5 의 코드가 local환경에서도 존재해야지 실행이 됩니다!
먼저 프로젝트 root경로에서 파일을 다운받아 줍니다.
git clone https://github.com/ultralytics/yolov5.git
clone받은 폴더안에 requirements.txt의 라이브러리들을 다운받아 줍니다.
cd yolov5
pip3 install -r requirements.txt
사진처럼 yolov5폴더 , model경로를 지정해주고 model을 load해주면 끝!