DY N DY

우분투 16.04.1에 darknet 설치 본문

PARK/영상처리 관련

우분투 16.04.1에 darknet 설치

손세지 2016. 10. 19. 22:55

딥러닝을 이용한 image classification을 시험해보기 위해 이것저것 찾던 중...

빠르고 가볍다는 yolo(http://pjreddie.com/darknet/yolo/)를 사용해 보기로 하였다. 

생초보라 이것저것 보면서 공부를 하기는 했는데... 사실 봐도봐도 모르겠으므로... 무작정 설치해보고 결과를 보며 신기해하는걸로...

yolo를 사용하기 위해서는 darknet을 설치해야 한다. 

darknet은 C와 CUDA로 짜여진 오픈소스 neural network framework이다.


관련 paper는 https://arxiv.org/pdf/1506.02640v5.pdf에서 확인 가능하다. 사실 능력이 안되는지라 읽어보기는 했는데 잘 모르겠다.. 신기방기 


우선 opencv를 설치한 후에(설치방법 : http://dyndy.tistory.com/243) 시작한다.

(없어도 사용할 수 있으나 이미지를 띄워서 boundingbox와 함께 보여주기 위해서는 opencv가 필요한 듯 하다. CUDA도 설치되어있으면 GPU를 이용해 더 빠른 연산이 가능하다고 하나 여기서는 하지 않았다.)


1. git에서 darknet을 clone한다. 

1
git clone https://github.com/pjreddie/darknet.git
cs


2. 만약 CUDA또는 opencv가 설치되어 있다면 사용하기 위해서 vi에디터 등으로 Makefile을 연다.

1
vi Makefile
cs


CUDA를 사용한다면 GPU=1로

opencv를 사용한다면 OPENCV=1로 변경해준다. 

(여기서는 opencv만 사용하였다.)


3. 다운받은 폴더로 이동하여 make한다.

1
2
cd darknet
make
cs



4. 설치는 끝. opencv를 사용한 경우 실제 그림을 띄워 확인해 본다. 

1
./darknet imtest data/eagle.jpg
cs

eagle.jpg는 darknet clone시 같이 받아진 예제 이미지로 이미지가 뜨는 것을 확인할 수 있다. 

사실 아래 값들은... 무슨 값인지는 잘... 그냥 신기할 따름..