잘 되는지 확인

연예인 사진 다운로드 → photo 폴더 안에 넣기

주피터 노트북 실행

jupyter notebook

New → Python3 → 파일명 torch_test

import torch # 파이토치 불러오기

파이토치 버전 확인

torch.__version__

모델 올리기

model = torch.hub.load('ultralytics/yolov5', 'yolov5s') # 네트워크에서 불러오기, yolov5m, yolov5x

이미지 불러오기

img = './girl.jpg' # photo폴더 내에 있음

예측 prediction

model.conf = 0.3
results = model(img)

결과 확인

results.print()