├── .gitignore ├── LICENSE ├── README.md ├── config ├── __init__.py ├── base.yml ├── resnet-transformer.yml ├── resnet_fpn_transformer.yml ├── vgg-convseq2seq.yml ├── vgg-seq2seq.yml └── vgg-transformer.yml ├── image ├── .keep ├── sample.png └── vietocr.jpg ├── setup.py ├── vietocr ├── __init__.py ├── loader │ ├── __init__.py │ ├── aug.py │ ├── dataloader.py │ └── dataloader_v1.py ├── model │ ├── __init__.py │ ├── backbone │ │ ├── __init__.py │ │ ├── cnn.py │ │ ├── resnet.py │ │ └── vgg.py │ ├── beam.py │ ├── seqmodel │ │ ├── __init__.py │ │ ├── convseq2seq.py │ │ ├── seq2seq.py │ │ └── transformer.py │ ├── trainer.py │ ├── transformerocr.py │ └── vocab.py ├── optim │ ├── __init__.py │ ├── labelsmoothingloss.py │ └── optim.py ├── predict.py ├── requirement.txt ├── tests │ ├── image │ │ ├── 001099025107.jpeg │ │ ├── 026301003919.jpeg │ │ ├── 036170002830.jpeg │ │ ├── 038078002355.jpeg │ │ ├── 038089010274.jpeg │ │ ├── 038144000109.jpeg │ │ ├── 060085000115.jpeg │ │ ├── 072183002222.jpeg │ │ ├── 079084000809.jpeg │ │ └── 079193002341.jpeg │ ├── sample.txt │ └── utest.py ├── tool │ ├── __init__.py │ ├── config.py │ ├── create_dataset.py │ ├── logger.py │ ├── predictor.py │ ├── translate.py │ └── utils.py └── train.py └── vietocr_gettingstart.ipynb /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/README.md -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /config/base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/config/base.yml -------------------------------------------------------------------------------- /config/resnet-transformer.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/config/resnet-transformer.yml -------------------------------------------------------------------------------- /config/resnet_fpn_transformer.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/config/resnet_fpn_transformer.yml -------------------------------------------------------------------------------- /config/vgg-convseq2seq.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/config/vgg-convseq2seq.yml -------------------------------------------------------------------------------- /config/vgg-seq2seq.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/config/vgg-seq2seq.yml -------------------------------------------------------------------------------- /config/vgg-transformer.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/config/vgg-transformer.yml -------------------------------------------------------------------------------- /image/.keep: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /image/sample.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/image/sample.png -------------------------------------------------------------------------------- /image/vietocr.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/image/vietocr.jpg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/setup.py -------------------------------------------------------------------------------- /vietocr/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /vietocr/loader/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /vietocr/loader/aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/loader/aug.py -------------------------------------------------------------------------------- /vietocr/loader/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/loader/dataloader.py -------------------------------------------------------------------------------- /vietocr/loader/dataloader_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/loader/dataloader_v1.py -------------------------------------------------------------------------------- /vietocr/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /vietocr/model/backbone/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /vietocr/model/backbone/cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/model/backbone/cnn.py -------------------------------------------------------------------------------- /vietocr/model/backbone/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/model/backbone/resnet.py -------------------------------------------------------------------------------- /vietocr/model/backbone/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/model/backbone/vgg.py -------------------------------------------------------------------------------- /vietocr/model/beam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/model/beam.py -------------------------------------------------------------------------------- /vietocr/model/seqmodel/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /vietocr/model/seqmodel/convseq2seq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/model/seqmodel/convseq2seq.py -------------------------------------------------------------------------------- /vietocr/model/seqmodel/seq2seq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/model/seqmodel/seq2seq.py -------------------------------------------------------------------------------- /vietocr/model/seqmodel/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/model/seqmodel/transformer.py -------------------------------------------------------------------------------- /vietocr/model/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/model/trainer.py -------------------------------------------------------------------------------- /vietocr/model/transformerocr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/model/transformerocr.py -------------------------------------------------------------------------------- /vietocr/model/vocab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/model/vocab.py -------------------------------------------------------------------------------- /vietocr/optim/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /vietocr/optim/labelsmoothingloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/optim/labelsmoothingloss.py -------------------------------------------------------------------------------- /vietocr/optim/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/optim/optim.py -------------------------------------------------------------------------------- /vietocr/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/predict.py -------------------------------------------------------------------------------- /vietocr/requirement.txt: -------------------------------------------------------------------------------- 1 | einops==0.2.0 2 | -------------------------------------------------------------------------------- /vietocr/tests/image/001099025107.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/tests/image/001099025107.jpeg -------------------------------------------------------------------------------- /vietocr/tests/image/026301003919.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/tests/image/026301003919.jpeg -------------------------------------------------------------------------------- /vietocr/tests/image/036170002830.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/tests/image/036170002830.jpeg -------------------------------------------------------------------------------- /vietocr/tests/image/038078002355.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/tests/image/038078002355.jpeg -------------------------------------------------------------------------------- /vietocr/tests/image/038089010274.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/tests/image/038089010274.jpeg -------------------------------------------------------------------------------- /vietocr/tests/image/038144000109.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/tests/image/038144000109.jpeg -------------------------------------------------------------------------------- /vietocr/tests/image/060085000115.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/tests/image/060085000115.jpeg -------------------------------------------------------------------------------- /vietocr/tests/image/072183002222.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/tests/image/072183002222.jpeg -------------------------------------------------------------------------------- /vietocr/tests/image/079084000809.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/tests/image/079084000809.jpeg -------------------------------------------------------------------------------- /vietocr/tests/image/079193002341.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/tests/image/079193002341.jpeg -------------------------------------------------------------------------------- /vietocr/tests/sample.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/tests/sample.txt -------------------------------------------------------------------------------- /vietocr/tests/utest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/tests/utest.py -------------------------------------------------------------------------------- /vietocr/tool/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /vietocr/tool/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/tool/config.py -------------------------------------------------------------------------------- /vietocr/tool/create_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/tool/create_dataset.py -------------------------------------------------------------------------------- /vietocr/tool/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/tool/logger.py -------------------------------------------------------------------------------- /vietocr/tool/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/tool/predictor.py -------------------------------------------------------------------------------- /vietocr/tool/translate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/tool/translate.py -------------------------------------------------------------------------------- /vietocr/tool/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/tool/utils.py -------------------------------------------------------------------------------- /vietocr/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr/train.py -------------------------------------------------------------------------------- /vietocr_gettingstart.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pbcquoc/vietocr/HEAD/vietocr_gettingstart.ipynb --------------------------------------------------------------------------------