RetinaNet을 이용한 Object Detection

참조 깃헙

https://github.com/fizyr/keras-retinanet

fizyr/keras-retinanet

 

 

1. 환경구축

1.1 conda create -n env_name python=3.6

1.2 python setup.py build_ext --inplace

1.3 requirement.txt에서 keras==2.2.4, tensorflow-gpu==1.14로 수정

1.4 pip install -r requirement.txt

1.5 pip install --user git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI

1.6 keras_retinanet/utils/keras_version.py에서 minimum_keras_version = 2, 2, 4로 수정

1.7 https://github.com/fizyr/keras-retinanet/releases에서  resnet50_coco_best_v2.1.0.h5 다운 받고 snapshots 폴더로 이동

 

 

2. 실행

2.1 examples폴더에서 ResNet50RetinaNet.py 실행

 

실행 결과