[Darknet/YOLOv3] Traffic light detection 신호등 인식01

 

작년에 자율주행차량 프로젝트를 진행하였습니다.

그 중에서 저는 신호등 인식(Traffic light detection) 관련 부분을 담당하였습니다.

Darknet/YOLOv3를 사용하여 프로젝트를 진행하였습니다.

 

개발환경

Windows WSL2, Jetson Nano (Ubuntu 18.04), Darknet/YOLOv3

 

이전의 WSL2 설치 글

https://pjoongq.tistory.com/entry/Windows-WSL-%EC%84%A4%EC%B9%98-%EB%B0%8F-%EC%84%A4%EC%A0%95

신호등 인식시 참고한 Github

https://github.com/berktepebag/Traffic-light-detection-with-YOLOv3-BOSCH-traffic-light-dataset

Darknet install 다크넷 설치

본인의 개발환경에 맞춰서 디렉토리 설정을 합니다.

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

설정한 디렉토리에서 darknet 파일을 git clone으로 해서 가져온 후 make를 해줍니다.

 

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

make가 끝나면 위의 코드를 입력하여 아래의 그림처럼 인식이 되는 것을 확인할 수 있습니다.

./ 의 디렉토리에서 darknet 파일을 실행합니다.

cfg/ 디렉토리에 yolov3.cfg 설정 파일과

./ 디렉토리의 yolov3.weights 파일을 불러와서

detect 명령어로 data/dog.jpg파일을 detection 합니다.