[ubuntu] ubuntu에서 YOLOV4 설치(Install Yolo v4 on Ubuntu)

yolov3의 관한 자료들은 인터넷을 보면 많이 찾아볼 수가 있다 하지만 yolov4에 설치에 관한 자료는 많이 없으므로 yolov4설치에 대해 알아보자

 

실제로 yolov4는 yolov3에 비해 많은 성능이 향상되었다고 합니다.

성능 향상에 대해 관련 자료를 찾아보고 싶으시면 이 링크를 이용해보세요  

 

keyog.tistory.com/31

YOLOv4 : Object detection의 최적의 속도와 정확도

저는 AlexeyAB의 Github에 올라온 YOLO V4를 설치 하였습니다.

https://github.com/AlexeyAB/darknet

AlexeyAB/darknet

 

설치 전 꼭 opencv를 설치해주셔야 합니다. opencv설치 포스팅은 저의 블로그에도 나와있으니 설치가 안되어있으신 분들은 참고하시면 됩니다.

twobeach.tistory.com/category/OpenCV/%EC%84%A4%EC%B9%98

'OpenCV/설치' 카테고리의 글 목록

먼저 설치 전 github에 있는 zip파일을 가지고오기위해 터미널창에서 명령어를 작성합니다.

git clone https://github.com/AlexeyAB/darknet.git

 

받아온 zip파일을 unzip명령어를 통해 압축을 풀어주고 darknet 디렉토리 안에 있는 Makefile을 열어서 저와같이 수정하시면 됩니다.

 

Makefile 수정 후

cudnn_half를 1로하는 이유는 학습시킬때 학습시간을 향상시켜준다고 해서 1로 한 것이므로 원하지 않으신다면 안하셔도 됩니다.

 

밑의 사진처럼 터미널창에서 darknet 디렉토리로 이동한 후 make명령어를 쳐주면 darknet파일과 so파일이 생성됩니다.

만약에 cudnn이나 cuda가 설치가 안되신 분들은 위의 설치과정에서 오류가 날수 있으니 밑은 주소를 참고해서 설치하시길 바랍니다.

 

webnautes.tistory.com/1428

Ubuntu 20.04에 CUDA Toolkit와 cuDNN, Tensorflow 설치하기

위의 과정에서 아무 문제가 없으시다면 먼저 훈련된 가중치(Weight)파일을 다운받아 테스트를 해주시면 됩니다.

가중치 파일은 용량때문에 첨부가 안되니 밑의 주소에서 받으시면 됩니다.(밑의 주소에서 다운로드가 안된다면 댓글로 이메일주소를 주시면 보내드리겠습니다.)

 

drive.google.com/open?id=1cewMfusmPjYWbrnuJRuKhPMwRe_b9PaT

yolov4.weights

가중치(Weight)파일까지 다운로드가 완료되면 darknet 디렉토리로 이동하여 예제를 실행하시면 됩니다.

 

./darknet detect cfg/yolov4.cfg yolov4.weights data/dog.jpg

---

※ 만약에 cuda메모리관련 오류

CUDA status Error: file: ./src/dark_cuda.c : () : line: 373 : build time: Oct 22 2020 - 13:44:05 CUDA Error: out of memory CUDA Error: out of memory: Success darknet: ./src/utils.c:325: error: Assertion `0' failed.

이런 오류가 난다면 yolov4.cfg파일을 수정하시면됩니다.

 

width=608
height=608

 

↓↓↓

 

width=416
height=416