[Yolov5] detect.py 모델추론 및 옵션변경

📌 학습한 결과 모델 (best_model.pt)을 이용해 결과확인하기

Colab으로 학습시킨 .pt모델을 직접 사진을 넣어 결과값을 얻어보는 과정입니다!

📌 Yolov5 Clone받기

#clone
git clone https://github.com/ultralytics/yolov5

#yolov5 폴더로 이동 
cd yolov5

📌 명령어 실행

python3 detect.py --source ../hello.jpg --weights ../model/best_model.pt --save-txt

yolov5폴더로 이동하면 detect.py파일이 있습니다!

경로에 맞게 명령어를 설정해주고 여러옵션을 주어 실행합니다.

📌 옵션

detect.py파일을 열어보면 여러가지 옵션들을 확인할 수 있습니다! (공식문서 참조)

--source : 테스트할 사진의 경로
--weights : 학습한 모델의 경로 (best_model.pt의 경로)
--save-txt : 결과를 저장 여부

📌 결과

결과는 yolov5/runs/detect/exp 에 저장됩니다.

저장경로

결과사진