yolov5 모델 선택

Question

챗지피티에서 나오는 코드를 복붙하여 위와 같은 yolov5 모델 실시간인식을 진행시켰는데 이 코드는 그냥 ultralytics에서 제공하는 딥러닝 모델을 이용한 것이고 저는 colab에서 roboflow에서 다운받은 이미지들을 딥러닝해 yolov5 모델을 따로 했는데 밑에 보이는 코드에서 ultralytics가 제공하는 모델이 아닌 제가 이미지를 딥러닝시킨 yolov5모델을 사용하는 방법은 뭘까요?참고로 colab에서 yolov5모델을 딥러닝하고 파이참에서 실행시키는 과정은 https://www.youtube.com/watch?v&#61;U-wkRQ8U3GE 를 따라했습니다.

누렁렁이 · Answer

line6의 torch.hub.load를 바꾸시면 됩니다. model &#61; torch.hub.load('/home/project/yolov5', 'custom', path&#61;'/home/project/yolov5/best.pt', source&#61;'local') 실제 모델 weight이 있는 경로로 입력해주시면 될 것 같습니다.&#43;&#43;) window이시면 os.path 사용을 추천드립니다. 아마 GPT한테 물어보면 잘 해줄거에요 출처 : https://github.com/ultralytics/yolov5/discussions/5872

yolov5 모델 선택

1개의 답변

Q&A 목록