This is a c++ project deploying the famous EAST scene text detector. It detects text from natural scene images. To combine this detector with scene text recognizer for a full pipeline, checkout my another repo.
The project is written in c++ using tensorflow computational framework. It is tested using tensorflow 1.4. Newer version should be ok too, but not tested. Please install:
-
Tensorflow
-
nsync project: https://github.com/google/nsync.git This is needed for building tensorflow.
-
opencv3.3
-
protobuf
-
eigen
Please check this project on how to build project using tensorflow with cmake: https://github.com/cjweeks/tensorflow-cmake It greatly helped the progress of building this project. When building tensorflow library, please be careful since we need to use opencv. Looks like there is still problem when including tensorflow and opencv together. It will make opencv unable to read image. Check out this issue: tensorflow/tensorflow#14267 The answer by allenlavoie solved my problem, so I paste it here:
"In the meantime, as long as you're not using any custom ops you can build libtensorflow_cc.so with bazel build --config=monolithic, which will condense everything together into one shared object (no libtensorflow_framework dependence) and seal off non-TensorFlow symbols. That shared object will have protocol buffer symbols."
cd build
cmake ..
make
It will create an excutable named DetectText in bin folder.
Download the pretrained detector model and put it in model/
./DetectText --detector_graph='model/Detector_model.pb'
--input_dir=/path/to/test/images/ --output_dir=/path/to/write/output/directory
The detector training is based on the public repo.
The python script for converting the trained model to the frozen graph will be provided. Basically the frozen graph output is all the decoded text boxes(before nonmax suppresion) instead of the score_map and geometry_map. I implemented a simple nonmax suppresion based on polygon IOU. Boost::geometry is used for polygon related calculation.
A pretrained frozen graph will be provided based on the original code in repo.
- EAST EAST: an efficient and accurate scene text detector paper.
- tensorflow-cmake, Tutorial of Building Project with tensorflow using cmake.
- Dafang He. The Penn State University. [email protected] http://personal.psu.edu/duh188/