-
질문 & 답변
카테고리
-
세부 분야
컴퓨터 비전
-
해결 여부
미해결
제 프로젝트에 어떤 모델을 사용해야 할지 질문 드립니다
22.04.01 11:23 작성 조회수 107
0
안녕하십니까 교수님
현재 저는 파이썬 프로젝트에 외부 카메라를 연결하여 실시간으로 물체를 디텍팅 하는 프로젝트를 수행중입니다
파이썬 프로젝트는 외부 카메라 2대를 사용해 가져온 영상을 Open CV 사용해 전처리 하고
해당 이미지에서 YOLOV5 모델을 이용해
객체를 탐지하여 해당객체를 다시 Open CV 사용 해 처리하는 루틴인데요
Open CV 와 YOLO는 잘 맞지 않는 다고 하여
현재는 YOLOV5 모델을 사용중이지만 향후 SSD 바꿔야 하나 하는 고민이 있습니다
1.혹시 저와 같은 케이스의 경우 SSD 모델을 사용하는것이 좀더 나을지
2.번외로 nvidia jetson 보드를 사용해볼까 하는 생각인데 이런 개발 보드를 사용하면 코랩에 비해 눈에 띄게 빠르게 학습을 시킬수 있을지 궁금합니다
답변을 작성해보세요.
0
권 철민
지식공유자2022.04.01
안녕하십니까,
음, 어떻게 opencv를 전처리와 후처리에만 사용하시면 YOLO V5를 이용하셔도 아무 문제가 없습니다.
Opencv와 YOLO가 수행속도가 느리다는 것은 OpenCV에서 제공하는 DNN 패키지의 Yolo 모듈의 예측 성능이 느리다는 의미 입니다. DNN 패키지를 꼭 사용해야 한다면 모르겠지만 그렇지 않다면
이미지 전처리를 OpenCV로 하고, 해당 이미지를 Yolo v5를 이용해 Detect하고 다시 OpenCV로 표출하면 될 것 같습니다.
nvidia jetson 보드에서 학습은 권장드리지 않습니다. 코랩등에서 먼저 학습한 모델을 jetson 보드로 이식하고 이 모델을 이용해 inference 하시면 어떨까 싶습니다. TensorRT가 jetson에서 지원되므로 이를 적용하면 좀 더 빠른 inference가 가능할 수 있습니다.
감사합니다.
답변 1