Yolo5를 이용하여 video object detection하기

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

이번시간에는 Yolov5를 이용하여 영상 속 객체를 탐지해보자

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

1. Yolov5 설치하기

https://github.com/ultralytics/yolov5

위 주소에 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