인프런 커뮤니티 질문&답변
작성한 질문수
Python 프로젝트에 적용하기에 가장 적합한 형태의 모델은 어떤 형식인가요?
작성
·
310
0
Open CV를 활용한 파이썬 PC 프로그램을 만들고자 하는데요
기존에는 classification 만 하려는 목적으로 .h5 형태의 모델을 사용했었는데
Object Detection 바꾸게 되었습니다
처음엔 YOLO 전이학습이 가장 쉽게 모델을 추출할수 있는것 같아서 YOLOV5 전이학습을 통해 나온 .pt 모델을 사용하려고 했는데 이방법을 어려울것 같다고 하셔서요
혹시 어떤 형태의 Object Detection 모델이 GPU기반 프로그램에서 Open CV 와 활용했을때 가장 효율이 좋고 쉽게 사용할수 있을까요?
또 전이학습을 통해 .pt 파일이 아닌 .weights 와 .cfg 를 추출하는 방법이 있을까요? 답변부탁 드리겠습니다 교수님..... 감사합니다!
답변 1
0
권 철민
지식공유자
안녕하십니까,
Opencv DNN은 모듈은 Tensorflow와 가장 호환이 잘 됩니다.
그런데 Opencv DNN은 수행 성능에 아래와 같은 이슈가 있습니다.
1. GPU 활용이 어렵습니다. GPU 활용이 안되는건 아닌데 매우 어렵습니다.
2. Tensorflow 1.x 모델만 호환이 됩니다. 2.x 모델과는 호환되지 않습니다.
결론적으로 Opencv DNN은 tensorflow 1.x 로 pretrained 된 모델과 가장 잘 맞습니다. 아마도 opencv DNN의 SSD 모델이 가장 잘 맞을 것입니다.
감사합니다.