• 카테고리

    질문 & 답변
  • 세부 분야

    컴퓨터 비전

  • 해결 여부

    미해결

Python 프로젝트에 적용하기에 가장 적합한 형태의 모델은 어떤 형식인가요?

22.01.12 18:59 작성 조회수 190

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 모델이 가장 잘 맞을 것입니다. 

감사합니다.