New → Python3 → 파일명 flask_photo

폴더 생성

photo\static\photos

폴더 내에 이미지 넣기

확인

torch_test 주피터노트북 열고

이미지 불러오기에서

img = './static/photos/cat_3.jpg'

import torch

yolo 엔진 불러오기

model = torch.hub.load('ultralytics/yolov5', 'yolov5s') # 학습된 모델 가져오기

되는지 확인

img_path = 'static/photos/cat_1.jpg'
results = model(img_path)
results.print()