윈도우 환경에서 YOLO로 실시간 객체탐지

 

 

YOLO는 you only look once 의 줄임말로 딥러닝에 관심이 있다면 한번쯤 들어봤을 법한 강력한 객체 인식 알고리즘이다.

 

https://pjreddie.com/darknet/yolo/

 

개발자의 홈페이지에 리눅스 환경에서 설치와 실행하는 방법이 자세히 나와있다. 

필자는 우분투가 설치된 컴퓨터가 있지만 윈도우 환경에서 실행해보고자 한다. 방법은 크게 2가지가 있는데 만약 C언어로 개발할 계획이면 visual studio에서 YOLO를 빌드하는 방법이 있고 (YOLO는 C를 기본으로 개발됬다.), 파이썬을 더 선호한다면 파이썬의 강력한 딥러닝 툴인 텐서플로우를 이용하는 방법이 있다. 여기서는 후자의 방법을 소개한다.

 

YOLO를 실행하는 환경을 개발자는 darknet 이라고 명명했는데, 어떤 고마운 개발자 분께서 이 darknet을 텐서플로우 환경에서 구동하는 darkflow 툴을 제작해주었다. (darknet + tensorflow = darkflow)

 

https://github.com/thtrieu/darkflow

 

위 깃헙에서 자세한 설치방법이 나와있는데 여기 말고 필자의 깃헙에서 간단한 예제코드와 예제파일들을 함께 다운받아도 된다.

 

https://github.com/Hydragon516/darkflow

반응형

설치 방법은 필자가 작성한 README.md를 참고하면 된다. 그리 어렵지는 않다.

 

 

만약 웹캠을 이용한 실시간 객체탐지를 하고싶다면 video.py를 실행하면 된다.