├── LICENSE ├── README.md ├── dataload ├── __init__.py ├── aug │ ├── __init__.py │ ├── augment.py │ └── warp_mls.py ├── dataAug.py └── loader.py ├── dataset.py ├── doc_dict.txt ├── images ├── VIPTR_SOTA.png └── hello_word.txt ├── model.py ├── modules ├── SVTR.py ├── VIPTRv1-T_ch.py ├── VIPTRv1.py ├── VIPTRv2.py ├── VIPTRv2T_ch.py ├── dctc_loss.py ├── feature_extraction.py ├── prediction.py ├── rec_sar_loss.py ├── sequence_modeling.py ├── stn_head.py ├── tps_spatial_transformer.py └── transformation.py ├── optimizer.py ├── scene_dict.txt ├── test_benchmark.py ├── train_benchmark.py └── utils.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/README.md -------------------------------------------------------------------------------- /dataload/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -*- -------------------------------------------------------------------------------- /dataload/aug/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/dataload/aug/__init__.py -------------------------------------------------------------------------------- /dataload/aug/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/dataload/aug/augment.py -------------------------------------------------------------------------------- /dataload/aug/warp_mls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/dataload/aug/warp_mls.py -------------------------------------------------------------------------------- /dataload/dataAug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/dataload/dataAug.py -------------------------------------------------------------------------------- /dataload/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/dataload/loader.py -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/dataset.py -------------------------------------------------------------------------------- /doc_dict.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/doc_dict.txt -------------------------------------------------------------------------------- /images/VIPTR_SOTA.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/images/VIPTR_SOTA.png -------------------------------------------------------------------------------- /images/hello_word.txt: -------------------------------------------------------------------------------- 1 | Hello World! 2 | -------------------------------------------------------------------------------- /model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/model.py -------------------------------------------------------------------------------- /modules/SVTR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/modules/SVTR.py -------------------------------------------------------------------------------- /modules/VIPTRv1-T_ch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/modules/VIPTRv1-T_ch.py -------------------------------------------------------------------------------- /modules/VIPTRv1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/modules/VIPTRv1.py -------------------------------------------------------------------------------- /modules/VIPTRv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/modules/VIPTRv2.py -------------------------------------------------------------------------------- /modules/VIPTRv2T_ch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/modules/VIPTRv2T_ch.py -------------------------------------------------------------------------------- /modules/dctc_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/modules/dctc_loss.py -------------------------------------------------------------------------------- /modules/feature_extraction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/modules/feature_extraction.py -------------------------------------------------------------------------------- /modules/prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/modules/prediction.py -------------------------------------------------------------------------------- /modules/rec_sar_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/modules/rec_sar_loss.py -------------------------------------------------------------------------------- /modules/sequence_modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/modules/sequence_modeling.py -------------------------------------------------------------------------------- /modules/stn_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/modules/stn_head.py -------------------------------------------------------------------------------- /modules/tps_spatial_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/modules/tps_spatial_transformer.py -------------------------------------------------------------------------------- /modules/transformation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/modules/transformation.py -------------------------------------------------------------------------------- /optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/optimizer.py -------------------------------------------------------------------------------- /scene_dict.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/scene_dict.txt -------------------------------------------------------------------------------- /test_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/test_benchmark.py -------------------------------------------------------------------------------- /train_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/train_benchmark.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyedekkun/VIPTR/HEAD/utils.py --------------------------------------------------------------------------------