제 프로젝트에 어떤 모델을 사용해야 할지 질문 드립니다

Question

안녕하십니까 교수님  현재 저는 파이썬 프로젝트에 외부 카메라를 연결하여 실시간으로 물체를 디텍팅 하는 프로젝트를 수행중입니다  파이썬 프로젝트는 외부 카메라 2대를 사용해 가져온 영상을 Open CV 사용해 전처리 하고 해당 이미지에서  YOLOV5  모델을 이용해    객체를  탐지하여 해당객체를 다시 Open CV 사용 해 처리하는 루틴인데요    Open CV  와 YOLO는 잘 맞지 않는 다고 하여  현재는 YOLOV5  모델을 사용중이지만  향후   SSD  바꿔야 하나 하는 고민이 있습니다     1.혹시 저와 같은 케이스의 경우   SSD 모델을  사용하는것이 좀더 나을지  2.번외로  nvidia jetson 보드를 사용해볼까 하는 생각인데   이런 개발 보드를 사용하면 코랩에 비해 눈에 띄게 빠르게 학습을 시킬수 있을지 궁금합니다

권 철민 · Answer

안녕하십니까,  음, 어떻게 opencv를 전처리와 후처리에만 사용하시면 YOLO V5를 이용하셔도 아무 문제가 없습니다.  Opencv와 YOLO가 수행속도가 느리다는 것은 OpenCV에서 제공하는 DNN 패키지의 Yolo 모듈의 예측 성능이 느리다는 의미 입니다.  DNN 패키지를 꼭 사용해야 한다면 모르겠지만 그렇지 않다면  이미지 전처리를 OpenCV로 하고, 해당 이미지를 Yolo v5를 이용해 Detect하고 다시 OpenCV로 표출하면 될 것 같습니다.  nvidia jetson 보드에서 학습은 권장드리지 않습니다. 코랩등에서 먼저 학습한 모델을 jetson 보드로 이식하고 이 모델을 이용해 inference 하시면 어떨까 싶습니다.  TensorRT가 jetson에서 지원되므로 이를 적용하면 좀 더 빠른 inference가 가능할 수 있습니다.  감사합니다.