├── README.md ├── config ├── SVTR-T-Ch-nostn.yaml ├── SVTR-T-Ch-repstn.yaml ├── SVTR-T-Ch-stn.yaml ├── SVTR-T-Ch.yaml ├── SVTR-T.yaml ├── densenetCRNN.yaml ├── lcnetCRNN.yaml ├── mobilev3CRNN.yaml ├── replcnetCRNN.yaml ├── repmobilev3CRNN.yaml ├── repvggCRNN-nostn.yaml └── repvggCRNN.yaml ├── convert_data_img_to_lmdb.py ├── dataload ├── __init__.py ├── aug │ ├── __init__.py │ ├── augment.py │ └── warp_mls.py ├── dataAug.py └── loader.py ├── infer.py ├── keyFile ├── keys.txt └── new_keys.txt ├── models ├── BuildModel.py ├── __init__.py ├── backbones │ ├── __init__.py │ ├── densenet.py │ ├── lcnet.py │ ├── mobilev3.py │ ├── rep_lcnet.py │ ├── rep_mobilev3.py │ ├── repvgg.py │ ├── repvggblock.py │ ├── stn_head.py │ ├── svtrNet.py │ └── tps_spatial_transformer.py ├── heads │ ├── __init__.py │ ├── convHead.py │ ├── lstmHead.py │ └── svtrHead.py ├── loss │ ├── CTCloss.py │ ├── KLDivloss.py │ └── __init__.py └── optimizer │ ├── __init__.py │ └── optimizer.py ├── train.py ├── train.sh ├── utils ├── __init__.py ├── label2tensor.py ├── logger.py └── tools.py └── val.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/README.md -------------------------------------------------------------------------------- /config/SVTR-T-Ch-nostn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/config/SVTR-T-Ch-nostn.yaml -------------------------------------------------------------------------------- /config/SVTR-T-Ch-repstn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/config/SVTR-T-Ch-repstn.yaml -------------------------------------------------------------------------------- /config/SVTR-T-Ch-stn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/config/SVTR-T-Ch-stn.yaml -------------------------------------------------------------------------------- /config/SVTR-T-Ch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/config/SVTR-T-Ch.yaml -------------------------------------------------------------------------------- /config/SVTR-T.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/config/SVTR-T.yaml -------------------------------------------------------------------------------- /config/densenetCRNN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/config/densenetCRNN.yaml -------------------------------------------------------------------------------- /config/lcnetCRNN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/config/lcnetCRNN.yaml -------------------------------------------------------------------------------- /config/mobilev3CRNN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/config/mobilev3CRNN.yaml -------------------------------------------------------------------------------- /config/replcnetCRNN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/config/replcnetCRNN.yaml -------------------------------------------------------------------------------- /config/repmobilev3CRNN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/config/repmobilev3CRNN.yaml -------------------------------------------------------------------------------- /config/repvggCRNN-nostn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/config/repvggCRNN-nostn.yaml -------------------------------------------------------------------------------- /config/repvggCRNN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/config/repvggCRNN.yaml -------------------------------------------------------------------------------- /convert_data_img_to_lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/convert_data_img_to_lmdb.py -------------------------------------------------------------------------------- /dataload/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -*- -------------------------------------------------------------------------------- /dataload/aug/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/dataload/aug/__init__.py -------------------------------------------------------------------------------- /dataload/aug/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/dataload/aug/augment.py -------------------------------------------------------------------------------- /dataload/aug/warp_mls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/dataload/aug/warp_mls.py -------------------------------------------------------------------------------- /dataload/dataAug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/dataload/dataAug.py -------------------------------------------------------------------------------- /dataload/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/dataload/loader.py -------------------------------------------------------------------------------- /infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/infer.py -------------------------------------------------------------------------------- /keyFile/keys.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/keyFile/keys.txt -------------------------------------------------------------------------------- /keyFile/new_keys.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/keyFile/new_keys.txt -------------------------------------------------------------------------------- /models/BuildModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/models/BuildModel.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -*- 2 | -------------------------------------------------------------------------------- /models/backbones/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -*- 2 | -------------------------------------------------------------------------------- /models/backbones/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/models/backbones/densenet.py -------------------------------------------------------------------------------- /models/backbones/lcnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/models/backbones/lcnet.py -------------------------------------------------------------------------------- /models/backbones/mobilev3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/models/backbones/mobilev3.py -------------------------------------------------------------------------------- /models/backbones/rep_lcnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/models/backbones/rep_lcnet.py -------------------------------------------------------------------------------- /models/backbones/rep_mobilev3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/models/backbones/rep_mobilev3.py -------------------------------------------------------------------------------- /models/backbones/repvgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/models/backbones/repvgg.py -------------------------------------------------------------------------------- /models/backbones/repvggblock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/models/backbones/repvggblock.py -------------------------------------------------------------------------------- /models/backbones/stn_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/models/backbones/stn_head.py -------------------------------------------------------------------------------- /models/backbones/svtrNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/models/backbones/svtrNet.py -------------------------------------------------------------------------------- /models/backbones/tps_spatial_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/models/backbones/tps_spatial_transformer.py -------------------------------------------------------------------------------- /models/heads/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -*- 2 | -------------------------------------------------------------------------------- /models/heads/convHead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/models/heads/convHead.py -------------------------------------------------------------------------------- /models/heads/lstmHead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/models/heads/lstmHead.py -------------------------------------------------------------------------------- /models/heads/svtrHead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/models/heads/svtrHead.py -------------------------------------------------------------------------------- /models/loss/CTCloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/models/loss/CTCloss.py -------------------------------------------------------------------------------- /models/loss/KLDivloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/models/loss/KLDivloss.py -------------------------------------------------------------------------------- /models/loss/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -*- 2 | -------------------------------------------------------------------------------- /models/optimizer/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -*- 2 | -------------------------------------------------------------------------------- /models/optimizer/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/models/optimizer/optimizer.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/train.sh -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -*- 2 | -------------------------------------------------------------------------------- /utils/label2tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/utils/label2tensor.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/utils/tools.py -------------------------------------------------------------------------------- /val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BADBADBADBOY/OCR-TextRecog/HEAD/val.py --------------------------------------------------------------------------------