[YOLOv9] ubuntu환경에서 yolov9 실시간 인식

 

yolov9이 출시 되었습니다. 원래 바로 포스팅하고 싶었는데 바빠서 이제 하네요ㅠ

yolov9은 yolov7과 동일한 사람이 개발하였습니다.

아래 사진을 보시면 yolov9성능이 더 뛰어난것을 관찰할 수 있습니다.

 

 

저의 컴퓨터 환경은 ubuntu 20.04.02 환경에서 진행하였습니다.

 

 

기본셋팅 환경은 아래와 같이 진행 하였습니다.

 

anaconda 가상환경(python=3.10.13)

CUDA=11.4

cudnn=8.5

pytorch=1.12.0

torchvision=0.13.0

 

 

yolov9의 github를 참고하였습니다. 

https://github.com/WongKinYiu/yolov9

GitHub - WongKinYiu/yolov9: Implementation of paper - YOLOv9: Learning What You Want to Learn Using Programmable Gradient Inform

 

 

우선 yolov9을 다운로드 받기위해 터미널창에 아래와 같이 입력합니다.

git clone https://github.com/WongKinYiu/yolov9.git 

 

입력하게 되면 yolov9이 다운로드 됩니다.

 

 

그리고 yolov9을 구동하는데 필요로하는 라이브러리를 다운받기위해 아래와 같이 입력합니다.

pip install -r requirements.txt

 

이렇게 라이브러리를 다운받았다면 yolov9을 구동할 수 있습니다.

 

 

하지만 그전에 yolov9를 구동하는데 필요한 가중치 파일을 다운받아야 합니다.

아래 사진처럼 yolov9-c.pt, yolov9-e.pt 등등 클릭하면 자동으로 다운됩니다.

그리고 yolov9폴더 안으로 이동시켜 줍니다.

 

 

 

 

웹캠을 이용하기 위해 아래와 같은 코드를 터미널창에 순서대로 입력합니다.

cd yolov9


python detect_dual.py --weight yolov9-c.pt --source 0

 

위 yolov9-c.pt 이외에도 다른 종류의 가중치 파일을 이용할 수 있습니다.

 

 

 

아래 사진처럼 실시간 객체인식을 할 수 있습니다.

 

구동 결과