├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── NOTICE ├── README.md ├── configs ├── abinet_finetune.yaml ├── abinet_pretrain_vision_model.yaml ├── pretrain_language_model.yaml ├── semimtr_finetune.yaml ├── semimtr_pretrain_vision_model.yaml └── template.yaml ├── data ├── DATA.md └── charset_36.txt ├── demo.py ├── figures ├── .DS_Store ├── abinet_model_architecture.svg ├── semimtr_consistency_regularization.svg └── semimtr_vision_pretraining.svg ├── main.py ├── notebook_demo.ipynb ├── requirements.txt ├── semimtr ├── __init__.py ├── callbacks │ ├── __init__.py │ └── callbacks.py ├── dataset │ ├── __init__.py │ ├── augmentation_pipelines.py │ ├── dataset.py │ ├── dataset_consistency_regularization.py │ ├── dataset_selfsupervised.py │ └── weighted_sampler.py ├── losses │ ├── __init__.py │ ├── consistency_regularization_loss.py │ ├── losses.py │ └── seqclr_loss.py ├── modules │ ├── __init__.py │ ├── attention.py │ ├── backbone.py │ ├── model.py │ ├── model_abinet.py │ ├── model_abinet_iter.py │ ├── model_alignment.py │ ├── model_fusion_consistency_regularization.py │ ├── model_fusion_teacher_student_ema.py │ ├── model_language.py │ ├── model_seqclr_vision.py │ ├── model_vision.py │ ├── projections.py │ ├── resnet.py │ ├── seqclr_proj.py │ └── transformer.py └── utils │ ├── __init__.py │ ├── test.py │ ├── transforms.py │ └── utils.py ├── setup.py └── tools ├── create_lmdb_dataset.py ├── crop_by_word_bb_syn90k.py ├── prepare_wikitext103.ipynb └── remove_labels_from_lmdb.py /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/LICENSE -------------------------------------------------------------------------------- /NOTICE: -------------------------------------------------------------------------------- 1 | Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. 2 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/README.md -------------------------------------------------------------------------------- /configs/abinet_finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/configs/abinet_finetune.yaml -------------------------------------------------------------------------------- /configs/abinet_pretrain_vision_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/configs/abinet_pretrain_vision_model.yaml -------------------------------------------------------------------------------- /configs/pretrain_language_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/configs/pretrain_language_model.yaml -------------------------------------------------------------------------------- /configs/semimtr_finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/configs/semimtr_finetune.yaml -------------------------------------------------------------------------------- /configs/semimtr_pretrain_vision_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/configs/semimtr_pretrain_vision_model.yaml -------------------------------------------------------------------------------- /configs/template.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/configs/template.yaml -------------------------------------------------------------------------------- /data/DATA.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/data/DATA.md -------------------------------------------------------------------------------- /data/charset_36.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/data/charset_36.txt -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/demo.py -------------------------------------------------------------------------------- /figures/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/figures/.DS_Store -------------------------------------------------------------------------------- /figures/abinet_model_architecture.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/figures/abinet_model_architecture.svg -------------------------------------------------------------------------------- /figures/semimtr_consistency_regularization.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/figures/semimtr_consistency_regularization.svg -------------------------------------------------------------------------------- /figures/semimtr_vision_pretraining.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/figures/semimtr_vision_pretraining.svg -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/main.py -------------------------------------------------------------------------------- /notebook_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/notebook_demo.ipynb -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/requirements.txt -------------------------------------------------------------------------------- /semimtr/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /semimtr/callbacks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /semimtr/callbacks/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/callbacks/callbacks.py -------------------------------------------------------------------------------- /semimtr/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /semimtr/dataset/augmentation_pipelines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/dataset/augmentation_pipelines.py -------------------------------------------------------------------------------- /semimtr/dataset/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/dataset/dataset.py -------------------------------------------------------------------------------- /semimtr/dataset/dataset_consistency_regularization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/dataset/dataset_consistency_regularization.py -------------------------------------------------------------------------------- /semimtr/dataset/dataset_selfsupervised.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/dataset/dataset_selfsupervised.py -------------------------------------------------------------------------------- /semimtr/dataset/weighted_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/dataset/weighted_sampler.py -------------------------------------------------------------------------------- /semimtr/losses/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /semimtr/losses/consistency_regularization_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/losses/consistency_regularization_loss.py -------------------------------------------------------------------------------- /semimtr/losses/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/losses/losses.py -------------------------------------------------------------------------------- /semimtr/losses/seqclr_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/losses/seqclr_loss.py -------------------------------------------------------------------------------- /semimtr/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /semimtr/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/modules/attention.py -------------------------------------------------------------------------------- /semimtr/modules/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/modules/backbone.py -------------------------------------------------------------------------------- /semimtr/modules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/modules/model.py -------------------------------------------------------------------------------- /semimtr/modules/model_abinet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/modules/model_abinet.py -------------------------------------------------------------------------------- /semimtr/modules/model_abinet_iter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/modules/model_abinet_iter.py -------------------------------------------------------------------------------- /semimtr/modules/model_alignment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/modules/model_alignment.py -------------------------------------------------------------------------------- /semimtr/modules/model_fusion_consistency_regularization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/modules/model_fusion_consistency_regularization.py -------------------------------------------------------------------------------- /semimtr/modules/model_fusion_teacher_student_ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/modules/model_fusion_teacher_student_ema.py -------------------------------------------------------------------------------- /semimtr/modules/model_language.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/modules/model_language.py -------------------------------------------------------------------------------- /semimtr/modules/model_seqclr_vision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/modules/model_seqclr_vision.py -------------------------------------------------------------------------------- /semimtr/modules/model_vision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/modules/model_vision.py -------------------------------------------------------------------------------- /semimtr/modules/projections.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/modules/projections.py -------------------------------------------------------------------------------- /semimtr/modules/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/modules/resnet.py -------------------------------------------------------------------------------- /semimtr/modules/seqclr_proj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/modules/seqclr_proj.py -------------------------------------------------------------------------------- /semimtr/modules/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/modules/transformer.py -------------------------------------------------------------------------------- /semimtr/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /semimtr/utils/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/utils/test.py -------------------------------------------------------------------------------- /semimtr/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/utils/transforms.py -------------------------------------------------------------------------------- /semimtr/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/semimtr/utils/utils.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/setup.py -------------------------------------------------------------------------------- /tools/create_lmdb_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/tools/create_lmdb_dataset.py -------------------------------------------------------------------------------- /tools/crop_by_word_bb_syn90k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/tools/crop_by_word_bb_syn90k.py -------------------------------------------------------------------------------- /tools/prepare_wikitext103.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/tools/prepare_wikitext103.ipynb -------------------------------------------------------------------------------- /tools/remove_labels_from_lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/semimtr-text-recognition/HEAD/tools/remove_labels_from_lmdb.py --------------------------------------------------------------------------------