[Yolov5] Custom Dataset으로 Yolov5 돌려보기 : Inference

섬지 2022. 10. 11. 11:10

https://minmiin.tistory.com/15

[Yolov5] Custom Dataset으로 Yolov5 돌려보기 : Train

이전 글에서는 Yolo 모델을 위한 데이터셋을 만들었다. https://minmiin.tistory.com/14 [Yolo] Custom Dataset으로 Yolo 돌려보기 : 데이터셋 준비 이전에 Custom Dataset을 통해 yolo를 학습하는 글을 쓴 적이..

minmiin.tistory.com

이전 글에 이어 Inference를 수행해 보자.

Inference에는 detect.py를 이용한다.

!python detect.py --weights /content/yolov5/runs/train/test/weights/best.pt --source /content/drive/MyDrive/Dataset/FaceMask/images/maksssksksss0.png

train 후 얻었던 weight와 사진 한 장을 source로 넣고 inference를 수행했다.

결과는 yolov5/runs/detect/프로젝트명 에 저장된다.