본문 바로가기
Tech Story/AI, ML, BigData

Deep Learning - (2) Ubuntu16.04 에서 Darknet 설치하기

Deep Learning - (2) Ubuntu16.04 에서 Darknet 설치하기

2018. 8. 10.


1. OpenCV 사전설치

2. Git에서 다크넷 Clone 및 설치

3. 결과확인 및 테스트



1. OpenCV 사전설치


기존 포스팅 확인 - http://e3jake.tistory.com/38



2. Git에서 다크넷 Clone 및 설치


git에서 darknet clone을 다운로드 한다

pi@pi-15U560-MFLGL:~$ git clone https://github.com/pjreddie/darknet.git


Makefile을 열어서 현재 설치된 노트북/데스크탑의 CPU/GPU환경을 설정해준다.

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

여기서는 opencv만 사용해서 해당 값만 변경해줬다.


pi@pi-15U560-MFLGL: $ cd darknet

pi@pi-15U560-MFLGL:~/darknet$ vi Makefile


GPU=0  <-- gpu 사용시

CUDNN=0

OPENCV=1   <----opencv 사용시

OPENMP=0

DEBUG=0


ARCH= -gencode arch=compute_30,code=sm_30 \

      -gencode arch=compute_35,code=sm_35 \

      -gencode arch=compute_50,code=[sm_50,compute_50] \

      -gencode arch=compute_52,code=[sm_52,compute_52]

#      -gencode arch=compute_20,code=[sm_20,sm_21] \ This one is deprecated?


# This is what I use, uncomment if you know your arch and want to specify

# ARCH= -gencode arch=compute_52,code=compute_52


VPATH=./src/:./examples

SLIB=libdarknet.so

ALIB=libdarknet.a

EXEC=darknet

OBJDIR=./obj/


CC=gcc

....



make를 이용하여 컴파일한다.

pi@pi-15U560-MFLGL:~/darknet$ make



3. 결과 확인 및 테스트


pi@pi-15U560-MFLGL:~/darknet$  

pi@pi-15U560-MFLGL:~/darknet$ ./darknet imtest data/eagle.jpg

L2 Norm: 372.007568

-0.002260 1.041666 0.952124


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


End of Posting