├── .gitignore ├── LICENSE ├── README.md ├── ablation-study-images ├── ablation-study-attn-tl.png └── ablation-study-tl-attn.png ├── dataset-demo ├── demo │ ├── 01-01.png │ ├── 01-02.png │ ├── 01-03.png │ ├── 01-04.png │ ├── 01-05.png │ ├── 02-01.png │ ├── 02-02.png │ ├── 02-03.png │ ├── 02-04.png │ └── 02-05.png └── gt.txt ├── datasets ├── README.md ├── iam_rwth_partitions │ ├── RWTH.iam_word_gt_final.test.thresh.txt │ ├── RWTH.iam_word_gt_final.train.thresh.txt │ └── RWTH.iam_word_gt_final.val.thresh.txt ├── process_iam.py └── process_imgur5k.py ├── model ├── LICENSE.md ├── create_lmdb_dataset.py ├── dataset.py ├── model.py ├── modules │ ├── feature_extraction.py │ ├── prediction.py │ ├── sequence_modeling.py │ └── transformation.py ├── pytorchtools.py ├── test.py ├── train.py └── utils.py └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/README.md -------------------------------------------------------------------------------- /ablation-study-images/ablation-study-attn-tl.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/ablation-study-images/ablation-study-attn-tl.png -------------------------------------------------------------------------------- /ablation-study-images/ablation-study-tl-attn.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/ablation-study-images/ablation-study-tl-attn.png -------------------------------------------------------------------------------- /dataset-demo/demo/01-01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/dataset-demo/demo/01-01.png -------------------------------------------------------------------------------- /dataset-demo/demo/01-02.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/dataset-demo/demo/01-02.png -------------------------------------------------------------------------------- /dataset-demo/demo/01-03.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/dataset-demo/demo/01-03.png -------------------------------------------------------------------------------- /dataset-demo/demo/01-04.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/dataset-demo/demo/01-04.png -------------------------------------------------------------------------------- /dataset-demo/demo/01-05.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/dataset-demo/demo/01-05.png -------------------------------------------------------------------------------- /dataset-demo/demo/02-01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/dataset-demo/demo/02-01.png -------------------------------------------------------------------------------- /dataset-demo/demo/02-02.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/dataset-demo/demo/02-02.png -------------------------------------------------------------------------------- /dataset-demo/demo/02-03.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/dataset-demo/demo/02-03.png -------------------------------------------------------------------------------- /dataset-demo/demo/02-04.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/dataset-demo/demo/02-04.png -------------------------------------------------------------------------------- /dataset-demo/demo/02-05.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/dataset-demo/demo/02-05.png -------------------------------------------------------------------------------- /dataset-demo/gt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/dataset-demo/gt.txt -------------------------------------------------------------------------------- /datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/datasets/README.md -------------------------------------------------------------------------------- /datasets/iam_rwth_partitions/RWTH.iam_word_gt_final.test.thresh.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/datasets/iam_rwth_partitions/RWTH.iam_word_gt_final.test.thresh.txt -------------------------------------------------------------------------------- /datasets/iam_rwth_partitions/RWTH.iam_word_gt_final.train.thresh.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/datasets/iam_rwth_partitions/RWTH.iam_word_gt_final.train.thresh.txt -------------------------------------------------------------------------------- /datasets/iam_rwth_partitions/RWTH.iam_word_gt_final.val.thresh.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/datasets/iam_rwth_partitions/RWTH.iam_word_gt_final.val.thresh.txt -------------------------------------------------------------------------------- /datasets/process_iam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/datasets/process_iam.py -------------------------------------------------------------------------------- /datasets/process_imgur5k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/datasets/process_imgur5k.py -------------------------------------------------------------------------------- /model/LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/model/LICENSE.md -------------------------------------------------------------------------------- /model/create_lmdb_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/model/create_lmdb_dataset.py -------------------------------------------------------------------------------- /model/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/model/dataset.py -------------------------------------------------------------------------------- /model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/model/model.py -------------------------------------------------------------------------------- /model/modules/feature_extraction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/model/modules/feature_extraction.py -------------------------------------------------------------------------------- /model/modules/prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/model/modules/prediction.py -------------------------------------------------------------------------------- /model/modules/sequence_modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/model/modules/sequence_modeling.py -------------------------------------------------------------------------------- /model/modules/transformation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/model/modules/transformation.py -------------------------------------------------------------------------------- /model/pytorchtools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/model/pytorchtools.py -------------------------------------------------------------------------------- /model/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/model/test.py -------------------------------------------------------------------------------- /model/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/model/train.py -------------------------------------------------------------------------------- /model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/model/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dmitrijsk/AttentionHTR/HEAD/requirements.txt --------------------------------------------------------------------------------