Skip to content

weinixuehao/tf_doc_localisation

Repository files navigation

tf_doc_localisation

Requirements

  1. tensorflow version==1.12.0
  2. PIL (pip install PIL)
  3. download part of the dataset,extract it to $MIDV_DATASET
  4. Edit tf_doc_localisation/receipt_dataset.py line 18
midv_dir = MIDV_DATASET ## for example: /data/your_download_directory/midv_500

how to train the network

  1. cd tf_doc_localisation
  2. mkdir data
  3. python synthesis_data.py
  4. python receipt_dataset.py
  5. python train.py

how to test

  1. python test.py

some test samples.

case 1

case1

case 2

case2

failure case

failure case

todo

  1. add evaluation code.
  2. 🚧 🚧 🚧 🚧 🚧

tflite_convert
--output_file=255X340_frozen_model.tflite
--graph_def_file=255X340_inference.pb
--input_arrays=IteratorGetNext
--output_arrays=heats_map_regression/pred_keypoints/BiasAdd

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages