Single Shot Text Detector with Regional Attention
Switch branches/tags
Nothing to show
Clone or download
BestSonny Merge pull request #24 from bado-lee/master
PriorBox min_size should be float to handle float values in the prototxt
Latest commit 174d452 Mar 12, 2018
Permalink
Failed to load latest commit information.
cmake :add codes Nov 3, 2017
examples :add codes Nov 3, 2017
include/caffe :add codes Nov 3, 2017
matlab :add codes Nov 3, 2017
python :add codes Nov 3, 2017
src PriorBox min_size should be float to handle float values in the prototxt Mar 12, 2018
tools :add codes Nov 3, 2017
.gitignore :add codes Nov 3, 2017
.travis.yml :add codes Nov 3, 2017
CMakeLists.txt :add codes Nov 3, 2017
CONTRIBUTING.md :add codes Nov 3, 2017
CONTRIBUTORS.md :add codes Nov 3, 2017
INSTALL.md :add codes Nov 3, 2017
LICENSE :add codes Nov 3, 2017
Makefile :add codes Nov 3, 2017
Makefile.config.example :add codes Nov 3, 2017
README.md Update README.md Jan 19, 2018
caffe.cloc :add codes Nov 3, 2017

README.md

License

Single Shot Text Detector with Regional Attention

Introduction

SSTD is initially described in our ICCV 2017 spotlight paper.

A third-party implementation of SSTD + Focal Loss. Thanks, Ho taek Han

If you find it useful in your research, please consider citing:

@inproceedings{panhe17singleshot,
      Title   = {Single Shot Text Detector with Regional Attention},
      Author  = {He, Pan and Huang, Weilin and He, Tong and Zhu, Qile and Qiao, Yu and Li, Xiaolin},
      Note    = {Proceedings of Internatioanl Conference on Computer Vision (ICCV)},
      Year    = {2017}
      }
@inproceedings{panhe16readText,
      Title   = {Reading Scene Text in Deep Convolutional Sequences},
      Author  = {He, Pan and Huang, Weilin and Qiao, Yu and Loy, Chen Change and Tang, Xiaoou},
      Note    = {Proceedings of AAAI Conference on Artificial Intelligence, (AAAI)},
      Year    = {2016}
      }
@inproceedings{liu16ssd,
      Title   = {{SSD}: Single Shot MultiBox Detector},
      Author  = {Liu, Wei and Anguelov, Dragomir and Erhan, Dumitru and Szegedy, Christian and Reed, Scott and Fu, Cheng-Yang and Berg, Alexander C.},
      Note    = {Proceedings of European Conference on Computer Vision (ECCV)},
      Year    = {2016}
      }

Installation

  1. Get the code. We will call the directory that you cloned Caffe into $CAFFE_ROOT
git clone https://github.com/BestSonny/SSTD.git
cd SSTD
  1. Build the code. Please follow Caffe instruction to install all necessary packages and build it.
# Modify Makefile.config according to your Caffe installation.
cp Makefile.config.example Makefile.config
make -j8
# Make sure to include $CAFFE_ROOT/python to your PYTHONPATH.
make py
make test -j8
# (Optional)
make runtest -j8
# build nms
cd examples/text
make
cd ..
  1. Run the demo code. Download Model google drive, baiduyun and put it in text/model folder
cd examples
sh text/download.sh
mkdir text/result
python text/demo_test.py