AdvancedEAST is an algorithm used for Scene image text detect, which is primarily based on EAST:An Efficient and Accurate Scene Text Detector, and the significant improvement was also made, which make long text predictions more accurate. If this project is helpful to you, welcome to star. And if you have any problem, please contact me.
- email:[email protected]
- wechat:gekongdianxue
- writen in keras, easy to read and run
- base on EAST, an advanced text detect algorithm
- easy to train the model
- significant improvement was made, long text predictions more accurate.(please see 'demo results' part bellow, and pay attention to the activation image, which starts with yellow grids, and ends with green grids.)
In my experiments,
AdvancedEast has obtained much better prediction accuracy then East,
especially on long text. Since East calculates final vertexes coordinates with
weighted mean values of predicted vertexes coordinates of all pixels. It is too
difficult to predict the 2 vertexes from the other side of the quadrangle.
See East limitations picked from original paper bellow.
- config file:cfg.py,control parameters
- pre-process data: preprocess.py,resize image
- label data: label.py,produce label info
- define network network.py
- define loss function losses.py
- execute training advanced_east.py and data_generator.py
- predict predict.py and nms.py
- AdvancedEast
- East
- python 3.6.3+
- tensorflow-gpu 1.5.0+(or tensorflow 1.5.0+)
- keras 2.1.4+
- numpy 1.14.1+
- tqdm 4.19.7+
- prepare training data:make data root dir(icpr), copy images to root dir, and copy txts to root dir, data format details could refer to 'ICPR MTWI 2018 挑战赛二：网络图像的文本检测', Link
- modify config params in cfg.py, see default values.
- python preprocess.py, resize image to 256256,384384,512512,640640,736*736, and train respectively could speed up training process.
- python label.py
- python advanced_east.py
- python predict.py -p demo/001.png, to predict
- compared with east based on vgg16
As you can see, although the text area prediction is very accurate, the vertex coordinates are not accurate enough.
The codes are released under the MIT License.
刚刚接触深度学习，有些地方理解还不够深入，可能会有一些错误，请大家多多包涵指正：）