Open instant-noodle-webcam.ipynb
with local jupyter notebook
- torch version 1.8.0 이상
- CUDA 11.1 이상
- Web Cam 필수
weights_path
: best.pt의 절대경로(폴더 위치: instant-noodle\yolov5\runs\train\noodle-result5\weights\best.pt)
Open instant_noodel_test_img.ipynb
with Colaboratory
- Google Drive
/content/drive/Mydrive
경로에instant-noodle
폴더 업로드 - 각 Cell 실행
- detect.py 실행 확인 방법
from IPython.display import Image
import os
val_img_path = img_list[47]
weights_path = '/content/drive/MyDrive/instant-noodle/yolov5/runs/train/noodle-result5/weights/best.pt'
!python detect.py --weights "{weights_path}" --img 416 --conf 0.5 --source "{val_img_path}"
detect_img_path = '/content/drive/MyDrive/instant-noodle/yolov5/runs/detect/exp'
위의 Cell 실행 후 'Results saved to runs/detect/exp'의 경로에 저장되었으므로, 해당 위치에서 확인 가능하다.
4. cv2_imshow 사용시 Results saved to runs/detect/**exp**/(이미지 파일 이름)
으로 작성해야 확인할 수 있다.
val_img_path
: test images의 경로를 저장한 listdetect_img_path
: test images 실행 결과 이미지를 저장한 폴더의 경로