yolo 설치 및 사용

opencv 설치 이후 yolo 설치를 진행한다. 터미널을 실행하여 다음의 명령어를 입력한다.

 

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

 

설치가 완료된 이후 darknet폴더가 홈에(바탕화면이 아님) 생성되는데  그 안의 Makefile을 실행하여 해당 부분을 변경해준다

 

CUDA 사용 GPU=1

opencv 사용  OPENCV=1

 

Makefile을 실행하는 방법은 여러가지가 있으므로 해당 내용만 변경후 저장해준다.

이후 변경이 완료되었으면 터미널에서 다음 명령어를 차례대로 실행한다.

 

cd darknet

make

이후 완료되었으면 yolo의 샘플 가중치 파일을 설치한다.  터미널을 통해 다운받을 경우 다음과 같이 입력한후 실행한다.

 

wget https://pjreddie.com/media/files/yolo.weights

*** 가중치의 파일이름과 현재 설치되어있는 yolo의 버전이 일치하는지 확인하여야 한다 틀리면 추후 실행시에 에러가 나는데 이를 해결하기 위해 2시간동안 구글링을 했던것 같다.

 

가중치 파일을 다운받았을 경우 다음 명령어를 실행한다.

 

./darknet detect cfg/yolo.cfg yolo.weights data/dog.jpg

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

 

//필자의 경우 v3 yolo를 사용하여 위와 같이 변경한 후 명령어를 입력하였다.

 

 

버전이 맞지 않을 경우 나오는 문구

 

 

 

실행이 정상적으로 진행될 경우 잠시 기다리면 다음과 같은 이미지 화면이 뜬다.

 

동영상 / 웹캠을 사용할시에도 아래의 명령어를 터미널에 입력한후 실행하면 demo창이 뜨며 영상 / 캠 이 동작한다.

 

./darknet detector demo cfg/coco.data cfg/yolov3.cfg yolov3.weights  <비디오 경로>