• 카테고리

    질문 & 답변
  • 세부 분야

    컴퓨터 비전

  • 해결 여부

    미해결

섹션 10. Ultralytics Yolo 결과물을 python 프로젝트에 적용하는 방법

22.01.12 11:24 작성 조회수 393

0

안녕하세요 교수님 먼저 좋은 강의 감사 드립니다 다름이 아니라 섹션 10. Ultralytics Yolo 실습 부분을 수강하고 Custom 데이터 사용해 학습까지 완료하였는데요

이렇게 나온 best.pt 와 last.pt 파일을 파이썬 open cv 프로젝트에 적용하고 싶습니다

기존에는 cv2.dnn.readNet("yolov3.weights", "yolov3.cfg") 이런식으로. weight 파일과 cfg 파일을 사용하거나 h5 사용하는것만 해보았는데

best.pt 와 last.pt 파일을 로컬에서 사용하는건 참고하기가 어렵더라구요. yolov3 의 detect.py 를 콘솔에서 사용하지 않고 cv2.dnn.readNet("yolov3.weights", "yolov3.cfg") 과 같은 형태로 쉽고 간단하게 사용할수는 없을까요?

best.pt 와 last.pt 파일 을 로컬 프로젝트에 적용하는 방법에 대해 참고할수 있는 방법 또는 , 참고자료, 교수님 강의중 참고할수있는 섹션 에 대해 알려주시면 감사하겠습니다!

답변 1

답변을 작성해보세요.

0

안녕하십니까, 

opencv dnn은 tensorflow만 weight를 지원합니다. 물론 ONNX 포맷으로 pytorch도 할수 있는 것 처럼 되어 있지만, (거의) 불가능 한걸로 알고 있습니다.