기억노트

Raspberry Pi darknet 본문

Programming

Raspberry Pi darknet

진바니 2020. 7. 9. 17:36

라즈베리파이 3에 darknet 설치하기

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

(darknet에는 pjreddie 버전과 alexey 버전이 있는데 yolo 논문의 저자가 만든 버전이 pjreddie 라고 한다.

먼저 pjreddie 버전을 설치해서 코드를 돌려보았는데, 

segmentation fault 떠서 삭제하고

alexey 버전으로 다시 설치했다.)

 

$ cd darknet

darknet 디렉토리로 들어간 다음,

 

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

로 yolo v3 tiny 의 weight 파일을 설치한다.

 

그 다음 make

$ make

(나는 pi 에서 openCV 없이 돌릴 예정이라 make 파일을 수정하지 않았다.)

 

설치가 완료되면 코드가 제대로 실행되는지 점검해본다.

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

위 코드를 돌리면 data 디렉토리에 저장되어 있는 dog.jpg 를 입력하여 결과를 사진으로 출력해준다.

출력 사진의 이름은 predictions.jpg 이고 gui 환경에서 사진을 확인하였다.