[Yolov5] .pt 모델 local환경에서 Load하기

홍영환·2023년 10월 15일
0

Yolov5

목록 보기
1/4
post-thumbnail

📌 Colab이 아닌 개인컴퓨터에서 .pt모델을 load하기

Colab으로 모델을 학습시킨 후 best_model.pt 파일을 이용해 병해충을 판별해주고 Flask서버를 통해서 결과를 넘겨주기로 하였다.

📌 .pt 모델 다운로드


Colab에서 .pt 모델만을 다운 받아 model/best_model.pt에 위치시킨 후 load를 해보았다.

📌 문제발생


app.py에서 위의 코드처럼 모델을 가져오고자 하였다.

하지만...

ModuleNotFoundError: No module named 'models'

models 모듈을 찾을 수 없다는 에러가 계속 발생하였다.

📌 해결방법

best_model.pt파일 + yolov5 파일

애초에 .pt파일을 실행시키려면
Yolov5 의 코드가 local환경에서도 존재해야지 실행이 됩니다!

  1. 먼저 프로젝트 root경로에서 파일을 다운받아 줍니다.
    git clone https://github.com/ultralytics/yolov5.git

  2. clone받은 폴더안에 requirements.txt의 라이브러리들을 다운받아 줍니다.
    cd yolov5
    pip3 install -r requirements.txt

  3. 사진처럼 yolov5폴더 , model경로를 지정해주고 model을 load해주면 끝!

📌 폴더 구조

profile
안녕하세요

0개의 댓글