[4] YOLO 설치 및 실행 방법

모든글의 설정 방식은 Ubuntu 18.04 버전에서 진행을 하였습니다.

 

YOLO 공식 사이트입니다.

https://pjreddie.com/darknet/yolo/

YOLO: Real-Time Object Detection

위의 공식  홈페이지에 나온 방법을 기준으로 설치를 진행하겠습니다.

1. git을 설치해줍니다.

$ sudo apt-get install git

 

2. darknet을 설치하고 make 해줍니다.

$ git clone https://github.com/pjreddie/darknet
$ cd darknet
$ make

 

3. Makefile 을 변경해줍니다.

sudo vi Makefile

위의 그림과 Makefile의 맨위 쪽에  OPENCV = 1 로 수정해줍니다.

만약 GPU를 사용하여 학습할시 GPU = 1

CUDNN을 이용할시 CUDNN = 1 이 됩니다.

Makefile은 수정후 반드시 make 를 해줘야 적응이 됩니다.

$ make

Makefile을 수정했으므로 make를 해줍니다.

 

4. 제대로 설치가 되었는지 확인을 위해 미리 학습된 가중치를 가져옵니다.(237MB)

 here (237 MB)

wget을 안하고 외부링크로 다운 받은후 darknet 폴더에 넣어주어도 됩니다.

$ wget https://pjreddie.com/media/files/yolov3.weights

 

5. 가져온 가중치를 돌려서 테스트 해봅니다.

$ ./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg

 

다음과 같은 결과가 나오면 yolo 를 성공적으로 설치한것입니다.

만약 Ubuntu 17.10이상의 버전을 사용하시면 다음과 같은 오류가 나올 수 있습니다.

./darknet: error while loading shared libraries: libopencv_highgui.so.3.2: cannot open shared object file: No such file or directory

이 부분을 해결하기 위해서는

$ sudo /bin/bash -c 'echo "/usr/local/lib" > /etc/ld.so.conf.d/opencv.conf'
$ sudo ldconfig

 

다음의 방법을 터미널에서 입력하면 됩니다.

오류 해결 참고

https://github.com/pjreddie/darknet/issues/382

Darknet Start Problem · Issue #382 · pjreddie/darknet