object_detection_yolo_darknet

ro_ot ㅣ 2020. 1. 22. 15:48

※ 물체 탐지(Object detection) - YOLO DarkNet

     *  YOLO의 원 코드 DarkNet으로 물체 탐지 실행.

▶ 프로젝트 다운로드

%cd /content

!rm -rf darknet

!git clone https://github.com/pjreddie/darknet
# 빌드
%cd darknet
!make

# 빌드된 실행파일 확인
!ls -al darknet

# 빌드 확인
!./darknet

▶ 학습된 모델 파일 다운로드

%cd /content/darknet
!wget wget https://pjreddie.com/media/files/yolov3.weights

▶ 물체 탐지 실행

     *  대상 파일은 data/dog.jpg

     *  모델의 웨이트 파일은 yolov3.weights

     *  모델 구조에 대한 설정 파일은 cfg/yolov3.cfg

     *  예측된 결과는 predictions.jpg 파일로 저장된다.

%cd /content/darknet
!./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg

from IPython.display import Image
Image('predictions.jpg')

▶ 다른 이미지에 물체 탐지

!wget -O street_view.jpg http://beltlineorg-wpengine.netdna-ssl.com/wp-content/uploads/2012/12/IMG_5669-flickr-websized.jpg

from IPython.display import Image, display
display(Image(filename='street_view.jpg'))
%cd /content/darknet
!./darknet detect cfg/yolov3.cfg yolov3.weights street_view.jpg

from IPython.display import Image
Image('predictions.jpg')

%cd /content/darknet
!./darknet detect cfg/yolov3.cfg yolov3.weights street_view.jpg

from IPython.display import Image
Image('predictions.jpg')

'Deep learning > Code' 카테고리의 다른 글

2020_01_23 KNN  (0) 2020.01.23
object_detection_yolo_keras  (2) 2020.01.22
2020_01_22 VGG16  (0) 2020.01.22
2020_01_22 CIFAR10  (0) 2020.01.22
2020_01_21 CNN  (0) 2020.01.21