OpenCV를 이용하여 video object detection하기(with Yolov5)

저번시간에 Pytorch를 GPU로 활성화 시키는 방법을 알아보았다.

 

이번시간에는 OpenCV를 이용해서 객체 탐지를 해보자.

추가로 Yolov5를 이용할 것이다.

 

저번 시간에 사용했던 가상환경 그대로 이어서 진행하자

 

1. Yolov5 설치하기

https://github.com/ultralytics/yolov5

GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

위 주소에 READ.ME를 보면 다음과 같은 명령어가 나와있다.

git clone https://github.com/ultralytics/yolov5  # clone
cd yolov5
pip install -r requirements.txt  # install

여기서 주의

우리는 저번 시간에 이미 Pytorch와 torchvision을 설치했다.

따라서 requirements.txt 파일에서 16, 17번째 줄을 주석처리해주자. (버전 충돌을 위해)

그 후 pip install -r requirements.txt 명령어로 설치를 진행한다.

 

2. streming Object detection

우리는 youtube video를 가져와서 스트리밍과 동시에 객체 탐지를 할 것이다.

#github에서 clone후, 다음 생성된 yolov5폴더로 이동
cd yolov5

#파일 실행
python detect.py --source https://youtu.be/7NQCcmKmm30

 

3. 결과 확인

 

reference

https://daddynkidsmakers.blogspot.com/2021/04/yolo-v5-install-and-run.html

https://github.com/ultralytics/yolov5