NVIDIA / caffe

Caffe: a fast open framework for deep learning.
http://caffe.berkeleyvision.org/
Other
672 stars 263 forks source link

Add RefineDet Object Detection #558

Closed cathy-kim closed 5 years ago

cathy-kim commented 5 years ago

Single-Shot Refinement Neural Network for Object Detection (https://arxiv.org/abs/1711.06897,https://github.com/sfzhang15/RefineDet) could improve mAP to 80.0%.

SSD is currently integrated into nvcaffe. However, RefineDet is not. It is not only compatible with SDD, but also available to train RefineDet, upgrade version of SSD.
It also includes code for FP16.

This pull request will enable RefineDet object detection in NVIDIA/caffe. (Any feedback for it would be grateful.)