![[Yolov5] Custom Dataset으로 Yolov5 돌려보기 : Inference](https://img1.daumcdn.net/thumb/R750x0/?scode=mtistory2&fname=https%3A%2F%2Fblog.kakaocdn.net%2Fdn%2FRFOO2%2FbtrOigUhgbW%2FXYBNfrfgk8l6llVDLkQQlK%2Fimg.png)
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/프로젝트명 에 저장된다.
'Deep Learning > Computer Vision' 카테고리의 다른 글
| [Yolov5] Custom Dataset으로 Yolov5 돌려보기 : Test (0) | 2022.10.11 |
|---|---|
| [Yolov5] Custom Dataset으로 Yolov5 돌려보기 : Train (1) | 2022.10.11 |
| [Yolov5] Custom Dataset으로 Yolov5 돌려보기 : 데이터셋 준비 (0) | 2022.10.11 |