[Machine Learning] Colab에 Darknet 설치하기

1. Darknet 설치하기

Colab에 맞게 환경설정이 적용된 Darknet 파일을 찾아야 하는데 https://github.com/kriyeng/darknet/ 를 이용하는 자료도 많다. 내 컴퓨터에서는 AlexeyAB 깃헙의 있는 darknet 파일이 잘 설치가 되었다.

!git clone https://github.com/AlexeyAB/darknet.git
%ls

%ls 명령어를 통해 content 폴더 밑에 darknet 폴더가 생성되어 있는 것을 확인하였다.

 

2. darknet 폴더로 이동 후 makefile 변경하기

%cd /content/darknet/
!sed -i 's/OPENCV=0/OPENCV=1/' Makefile
!sed -i 's/GPU=0/GPU=1/' Makefile
!sed -i 's/CUDNN=0/CUDNN=1/' Makefile
!sed -i 's/CUDNN_HALF=0/CUDNN_HALF=1/' Makefile

해당 명령어를 실행하고 Makefile을 찾아 상수가 변경된 것을 확인하였다.

 

 

3. 컴파일

!make

 

4. darknet 실행파일(./darknet)을 둘 폴더를 구글 드라이브에 생성

import os
if not os.path.exists('/content/gdrive/MyDrive/darknet/bin'): 
	os.makedirs('/content/gdrive/MyDrive/darknet/bin')

 

 

5. darknet 실행파일(./darknet)을 구글 드라이브로 복사

Colab 세션이 끝난 이후 다시 notebook 을 수행시킬 때 darknet 실행파을을 가져올 수 있도록 구글 드라이브에 백업(cp)한 후 실행 권한을 추가(chmod +x)한다.

%cp ./darknet /content/gdrive/MyDrive/darknet/bin/darknet
!chmod +x ./darknet

 

해당 경로에 darknet 이 생성된 것을 확인하였다.