YOLO V4를 이용한 Object Detection - (2) 영상 내 객체 검출

Intro

동영상은 연속된 이미지로 이루어져 있으므로 영상을 프레임 단위로 잘라서 각 프레임마다 객체를 검출하고 결과 프레임을 이어붙이면 영상 내 객체 검출을 할 수 있다. 이 과정을 opencv-python을 이용하여 진행했다.

---

opencv-python 패키지에 readNetFromDarknet()함수를 통해 darknet모델을 python에서 사용할 수 있었다. 

모델은 이전 포스팅에서 사용한 것과 동일하다. (coco데이터셋으로 학습된 yolov4)  

 

동영상에서 object detection한 결과로 검출된 객체에 bounding box를 표시하고 해당 객체의 label과 confidence를 함께 표시한다. 그 결과를 avi파일로 저장한다. 

 

코드는 아래 링크를 참고했다.

www.pyimagesearch.com/2018/11/12/yolo-object-detection-with-opencv/

YOLO object detection with OpenCV - PyImageSearch