※ 물체 탐지(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 |