qf6101 / DifferentiableBinarization

DB (Real-time Scene Text Detection with Differentiable Binarization) implementation in Keras and Tensorflow

DifferentiableBinarization

This is an implementation of Real-time Scene Text Detection with Differentiable Binarization in Keras and Tensorflow, Most portions of code are borrowed from the official implementation MhLiao/DB.

Build Dataset

Build dataset in the same way as the official implementation.

  datasets/total_text/train_images
  datasets/total_text/train_gts
  datasets/total_text/train_list.txt
  datasets/total_text/test_images
  datasets/total_text/test_gts
  datasets/total_text/test_list.txt

Train

python train.py Here is a trained model on TotalText baidu netdisk code:jy6m

Test

python inference.py