[Object Detection] Ubuntu에서 YOLOv5 실행하기

⭐ 본 포스팅에서는 ubuntu 환경에서 yolov5 환경 설정하고 돌려보겠습니다!

 

실습은 아래의 코드로 진행됩니다. 

https://github.com/ultralytics/yolov5

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

 

0. 환경 

ubuntu18.04 에서 진행했으며, anaconda를 이용합니다. 

 

1. conda yolov5 가상 환경 생성 

• 터미널을 열어서 가상환경을 생성

conda create -n yolov5 python=3.8

conda activate yolov5

 

 

2.  git clone 

• github repository를 clone

git clone https://github.com/ultralytics/yolov5.git
cd yolov5

 

3. requirement 설치 

pip install -r requirements.txt

 

4. 실행시키기! 

• 파일에 들어있던 이미지로 실행시켜보기

python detect.py --source data/images/bus.jpg

• 결과물은 runs/detect/exp 폴더에 저장된다. 

이와 같은 사진이 폴더 아래 저장되어 있다면 성공!

• 웹캠으로 실행 시키는 법

python detect.py --source 0

• 아래와 같은 다양한 방법으로도 실행이 가능합니다!

python detect.py --source 0  # webcam
                          img.jpg  # image
                          vid.mp4  # video
                          path/  # directory
                          path/*.jpg  # glob
                          'https://youtu.be/Zgi9g1ksQHc'  # YouTube
                          'rtsp://example.com/media.mp4'  # RTSP, RTMP, HTTP stream