├── README.md ├── Tutorial.ipynb ├── conf ├── config.yaml ├── dataset │ └── dmec.yaml ├── lm_models │ └── n_gram.yaml ├── model │ ├── vgg_conformer.yaml │ └── vgg_transformer.yaml ├── optimizer │ └── adam.yaml └── pl_params │ └── pl_params.yaml ├── requirements └── requirements.txt ├── scripts ├── train.py └── train_PT.py ├── test_img.png ├── transformer_ocr ├── core │ ├── activations.py │ ├── attentions.py │ ├── conformer_modules.py │ ├── embeddings.py │ └── optimizers.py ├── models │ ├── cnn_extraction │ │ ├── feature_extraction.py │ │ └── vgg.py │ ├── model.py │ ├── model_PT.py │ └── transformers │ │ ├── conformer.py │ │ └── tr_encoder.py └── utils │ ├── augment.py │ ├── config.py │ ├── dataset.py │ ├── image_processing.py │ ├── logger.py │ ├── metrics.py │ └── vocab.py └── visualization └── architecture.png /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/README.md -------------------------------------------------------------------------------- /Tutorial.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/Tutorial.ipynb -------------------------------------------------------------------------------- /conf/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/conf/config.yaml -------------------------------------------------------------------------------- /conf/dataset/dmec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/conf/dataset/dmec.yaml -------------------------------------------------------------------------------- /conf/lm_models/n_gram.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/conf/lm_models/n_gram.yaml -------------------------------------------------------------------------------- /conf/model/vgg_conformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/conf/model/vgg_conformer.yaml -------------------------------------------------------------------------------- /conf/model/vgg_transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/conf/model/vgg_transformer.yaml -------------------------------------------------------------------------------- /conf/optimizer/adam.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/conf/optimizer/adam.yaml -------------------------------------------------------------------------------- /conf/pl_params/pl_params.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/conf/pl_params/pl_params.yaml -------------------------------------------------------------------------------- /requirements/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/requirements/requirements.txt -------------------------------------------------------------------------------- /scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/scripts/train.py -------------------------------------------------------------------------------- /scripts/train_PT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/scripts/train_PT.py -------------------------------------------------------------------------------- /test_img.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/test_img.png -------------------------------------------------------------------------------- /transformer_ocr/core/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/transformer_ocr/core/activations.py -------------------------------------------------------------------------------- /transformer_ocr/core/attentions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/transformer_ocr/core/attentions.py -------------------------------------------------------------------------------- /transformer_ocr/core/conformer_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/transformer_ocr/core/conformer_modules.py -------------------------------------------------------------------------------- /transformer_ocr/core/embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/transformer_ocr/core/embeddings.py -------------------------------------------------------------------------------- /transformer_ocr/core/optimizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/transformer_ocr/core/optimizers.py -------------------------------------------------------------------------------- /transformer_ocr/models/cnn_extraction/feature_extraction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/transformer_ocr/models/cnn_extraction/feature_extraction.py -------------------------------------------------------------------------------- /transformer_ocr/models/cnn_extraction/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/transformer_ocr/models/cnn_extraction/vgg.py -------------------------------------------------------------------------------- /transformer_ocr/models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/transformer_ocr/models/model.py -------------------------------------------------------------------------------- /transformer_ocr/models/model_PT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/transformer_ocr/models/model_PT.py -------------------------------------------------------------------------------- /transformer_ocr/models/transformers/conformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/transformer_ocr/models/transformers/conformer.py -------------------------------------------------------------------------------- /transformer_ocr/models/transformers/tr_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/transformer_ocr/models/transformers/tr_encoder.py -------------------------------------------------------------------------------- /transformer_ocr/utils/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/transformer_ocr/utils/augment.py -------------------------------------------------------------------------------- /transformer_ocr/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/transformer_ocr/utils/config.py -------------------------------------------------------------------------------- /transformer_ocr/utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/transformer_ocr/utils/dataset.py -------------------------------------------------------------------------------- /transformer_ocr/utils/image_processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/transformer_ocr/utils/image_processing.py -------------------------------------------------------------------------------- /transformer_ocr/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/transformer_ocr/utils/logger.py -------------------------------------------------------------------------------- /transformer_ocr/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/transformer_ocr/utils/metrics.py -------------------------------------------------------------------------------- /transformer_ocr/utils/vocab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/transformer_ocr/utils/vocab.py -------------------------------------------------------------------------------- /visualization/architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoangtuanvu/conformer_ocr/HEAD/visualization/architecture.png --------------------------------------------------------------------------------