← all repositories

MichaelHL-ai/tensorflow_PSENet

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

487 stars C++ Computer Vision
tensorflow_PSENet
Velocity · 7d
+0.2
★ / day
Trend
steady
star history

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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.