- Notifications
You must be signed in to change notification settings - Fork78
Single Shot Text Detector with Regional Attention
License
NotificationsYou must be signed in to change notification settings
BestSonny/SSTD
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
SSTD is initially described in ourICCV 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} }- Get the code. We will call the directory that you cloned Caffe into
$CAFFE_ROOT
git clone https://github.com/BestSonny/SSTD.gitcd SSTD- Build the code. Please followCaffe instruction to install all necessary packages and build it.
# Modify Makefile.config according to your Caffe installation.cp Makefile.config.example Makefile.configmake -j8# Make sure to include $CAFFE_ROOT/python to your PYTHONPATH.make pymaketest -j8# (Optional)make runtest -j8# build nmscd examples/textmakecd ..
- Run the demo code. Download Modelgoogle drive,baiduyun and put it in
text/modelfolder
cd examplessh text/download.shmkdir text/resultpython text/demo_test.py