MichaelHL-ai/tensorflow_PSENet
A TensorFlow re-implementation of PSENet, a deep learning model for shape-robust text detection in natural scene images.

This project provides a TensorFlow implementation of the Progressive Scale Expansion Network (PSENet) for text detection. It uses a ResNet backbone and features a Progressive Scale Expansion algorithm to detect text of various scales and shapes. The implementation includes training and evaluation scripts for ICDAR datasets, with C++ acceleration for the PSE algorithm. Users can fine-tune on custom datasets and the model supports multi-GPU training.