xuannianz / 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 You can download a trained model weights on TotalText by baidu netdisk code:jy6m or google driver

Test

python inference.py