├── .gitignore ├── LICENSE ├── README.md ├── configs ├── dataset │ ├── icd13.yaml │ ├── locr.yaml │ ├── st.yaml │ └── tsg.yaml ├── demo.yaml ├── pretrain.yaml ├── test.yaml ├── test │ └── textdesign_sd_2.yaml ├── train.yaml └── train │ └── textdesign_sd_2.yaml ├── dataset ├── __init__.py ├── dataloader.py └── utils │ ├── arial.ttf │ └── words.txt ├── demo.py ├── demo ├── demo.png ├── examples │ ├── CEFUL_1_0.jpeg │ ├── CLOTHES_0_0.png │ ├── COMPLICATED_0_1.jpeg │ ├── DELIGHT_0_1.jpeg │ ├── ECHOES_0_0.jpeg │ ├── ENGINE_0_0.png │ ├── FASCINATING_0_1.jpeg │ ├── FAVOURITE_0_0.jpeg │ ├── FINNAL_0_1.jpeg │ ├── FRONTIER_0_0.png │ ├── Innovate_0_0.jpeg │ ├── PRESERVE_0_0.jpeg │ ├── Peaceful_0_0.jpeg │ ├── Scamps_0_0.png │ ├── TREE_0_0.png │ ├── better_0_0.jpg │ └── tested_0_0.png └── teaser.png ├── metrics.py ├── pretrain.py ├── requirements.txt ├── scripts └── preprocess │ └── laion_ocr_pre.ipynb ├── sgm ├── __init__.py ├── lr_scheduler.py ├── models │ ├── __init__.py │ ├── autoencoder.py │ └── diffusion.py ├── modules │ ├── __init__.py │ ├── attention.py │ ├── autoencoding │ │ ├── __init__.py │ │ ├── losses │ │ │ └── __init__.py │ │ └── regularizers │ │ │ └── __init__.py │ ├── diffusionmodules │ │ ├── __init__.py │ │ ├── denoiser.py │ │ ├── denoiser_scaling.py │ │ ├── denoiser_weighting.py │ │ ├── discretizer.py │ │ ├── guiders.py │ │ ├── loss.py │ │ ├── model.py │ │ ├── openaimodel.py │ │ ├── sampling.py │ │ ├── sampling_utils.py │ │ ├── sigma_sampling.py │ │ ├── util.py │ │ └── wrappers.py │ ├── distributions │ │ ├── __init__.py │ │ └── distributions.py │ ├── ema.py │ ├── encoders │ │ ├── __init__.py │ │ └── modules.py │ └── predictors │ │ └── model.py └── util.py ├── src └── parseq │ ├── .github │ ├── contexts-example.png │ ├── gh-teaser.png │ └── system.png │ ├── .gitignore │ ├── Datasets.md │ ├── LICENSE │ ├── NOTICE │ ├── README.md │ ├── bench.py │ ├── configs │ ├── bench.yaml │ ├── charset │ │ ├── 36_lowercase.yaml │ │ ├── 62_mixed-case.yaml │ │ └── 94_full.yaml │ ├── dataset │ │ ├── real.yaml │ │ └── synth.yaml │ ├── experiment │ │ ├── abinet-sv.yaml │ │ ├── abinet.yaml │ │ ├── crnn.yaml │ │ ├── parseq-patch16-224.yaml │ │ ├── parseq-tiny.yaml │ │ ├── parseq.yaml │ │ ├── trba.yaml │ │ ├── trbc.yaml │ │ ├── tune_abinet-lm.yaml │ │ └── vitstr.yaml │ ├── main.yaml │ ├── model │ │ ├── abinet.yaml │ │ ├── crnn.yaml │ │ ├── parseq.yaml │ │ ├── trba.yaml │ │ └── vitstr.yaml │ └── tune.yaml │ ├── demo_images │ ├── art-01107.jpg │ ├── coco-1166773.jpg │ ├── cute-184.jpg │ ├── ic13_word_256.png │ ├── ic15_word_26.png │ └── uber-27491.jpg │ ├── hubconf.py │ ├── read.py │ ├── requirements.txt │ ├── setup.cfg │ ├── setup.py │ ├── strhub │ ├── __init__.py │ ├── data │ │ ├── __init__.py │ │ ├── aa_overrides.py │ │ ├── augment.py │ │ ├── dataset.py │ │ ├── module.py │ │ └── utils.py │ └── models │ │ ├── __init__.py │ │ ├── abinet │ │ ├── LICENSE │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── backbone.py │ │ ├── model.py │ │ ├── model_abinet_iter.py │ │ ├── model_alignment.py │ │ ├── model_language.py │ │ ├── model_vision.py │ │ ├── resnet.py │ │ ├── system.py │ │ └── transformer.py │ │ ├── base.py │ │ ├── crnn │ │ ├── LICENSE │ │ ├── __init__.py │ │ ├── model.py │ │ └── system.py │ │ ├── modules.py │ │ ├── parseq │ │ ├── __init__.py │ │ ├── modules.py │ │ └── system.py │ │ ├── trba │ │ ├── __init__.py │ │ ├── feature_extraction.py │ │ ├── model.py │ │ ├── prediction.py │ │ ├── system.py │ │ └── transformation.py │ │ ├── utils.py │ │ └── vitstr │ │ ├── __init__.py │ │ ├── model.py │ │ └── system.py │ ├── test.py │ ├── tools │ ├── art_converter.py │ ├── case_sensitive_str_datasets_converter.py │ ├── coco_2_converter.py │ ├── coco_text_converter.py │ ├── create_lmdb_dataset.py │ ├── filter_lmdb.py │ ├── lsvt_converter.py │ ├── mlt19_converter.py │ ├── openvino_converter.py │ ├── test_abinet_lm_acc.py │ └── textocr_converter.py │ ├── train.py │ └── tune.py ├── test.py ├── train.py └── util.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/README.md -------------------------------------------------------------------------------- /configs/dataset/icd13.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/configs/dataset/icd13.yaml -------------------------------------------------------------------------------- /configs/dataset/locr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/configs/dataset/locr.yaml -------------------------------------------------------------------------------- /configs/dataset/st.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/configs/dataset/st.yaml -------------------------------------------------------------------------------- /configs/dataset/tsg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/configs/dataset/tsg.yaml -------------------------------------------------------------------------------- /configs/demo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/configs/demo.yaml -------------------------------------------------------------------------------- /configs/pretrain.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/configs/pretrain.yaml -------------------------------------------------------------------------------- /configs/test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/configs/test.yaml -------------------------------------------------------------------------------- /configs/test/textdesign_sd_2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/configs/test/textdesign_sd_2.yaml -------------------------------------------------------------------------------- /configs/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/configs/train.yaml -------------------------------------------------------------------------------- /configs/train/textdesign_sd_2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/configs/train/textdesign_sd_2.yaml -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/dataset/dataloader.py -------------------------------------------------------------------------------- /dataset/utils/arial.ttf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/dataset/utils/arial.ttf -------------------------------------------------------------------------------- /dataset/utils/words.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/dataset/utils/words.txt -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/demo.py -------------------------------------------------------------------------------- /demo/demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/demo/demo.png -------------------------------------------------------------------------------- /demo/examples/CEFUL_1_0.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/demo/examples/CEFUL_1_0.jpeg -------------------------------------------------------------------------------- /demo/examples/CLOTHES_0_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/demo/examples/CLOTHES_0_0.png -------------------------------------------------------------------------------- /demo/examples/COMPLICATED_0_1.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/demo/examples/COMPLICATED_0_1.jpeg -------------------------------------------------------------------------------- /demo/examples/DELIGHT_0_1.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/demo/examples/DELIGHT_0_1.jpeg -------------------------------------------------------------------------------- /demo/examples/ECHOES_0_0.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/demo/examples/ECHOES_0_0.jpeg -------------------------------------------------------------------------------- /demo/examples/ENGINE_0_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/demo/examples/ENGINE_0_0.png -------------------------------------------------------------------------------- /demo/examples/FASCINATING_0_1.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/demo/examples/FASCINATING_0_1.jpeg -------------------------------------------------------------------------------- /demo/examples/FAVOURITE_0_0.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/demo/examples/FAVOURITE_0_0.jpeg -------------------------------------------------------------------------------- /demo/examples/FINNAL_0_1.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/demo/examples/FINNAL_0_1.jpeg -------------------------------------------------------------------------------- /demo/examples/FRONTIER_0_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/demo/examples/FRONTIER_0_0.png -------------------------------------------------------------------------------- /demo/examples/Innovate_0_0.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/demo/examples/Innovate_0_0.jpeg -------------------------------------------------------------------------------- /demo/examples/PRESERVE_0_0.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/demo/examples/PRESERVE_0_0.jpeg -------------------------------------------------------------------------------- /demo/examples/Peaceful_0_0.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/demo/examples/Peaceful_0_0.jpeg -------------------------------------------------------------------------------- /demo/examples/Scamps_0_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/demo/examples/Scamps_0_0.png -------------------------------------------------------------------------------- /demo/examples/TREE_0_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/demo/examples/TREE_0_0.png -------------------------------------------------------------------------------- /demo/examples/better_0_0.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/demo/examples/better_0_0.jpg -------------------------------------------------------------------------------- /demo/examples/tested_0_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/demo/examples/tested_0_0.png -------------------------------------------------------------------------------- /demo/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/demo/teaser.png -------------------------------------------------------------------------------- /metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/metrics.py -------------------------------------------------------------------------------- /pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/pretrain.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/preprocess/laion_ocr_pre.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/scripts/preprocess/laion_ocr_pre.ipynb -------------------------------------------------------------------------------- /sgm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/__init__.py -------------------------------------------------------------------------------- /sgm/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/lr_scheduler.py -------------------------------------------------------------------------------- /sgm/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/models/__init__.py -------------------------------------------------------------------------------- /sgm/models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/models/autoencoder.py -------------------------------------------------------------------------------- /sgm/models/diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/models/diffusion.py -------------------------------------------------------------------------------- /sgm/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/modules/__init__.py -------------------------------------------------------------------------------- /sgm/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/modules/attention.py -------------------------------------------------------------------------------- /sgm/modules/autoencoding/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sgm/modules/autoencoding/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/modules/autoencoding/losses/__init__.py -------------------------------------------------------------------------------- /sgm/modules/autoencoding/regularizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/modules/autoencoding/regularizers/__init__.py -------------------------------------------------------------------------------- /sgm/modules/diffusionmodules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/modules/diffusionmodules/__init__.py -------------------------------------------------------------------------------- /sgm/modules/diffusionmodules/denoiser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/modules/diffusionmodules/denoiser.py -------------------------------------------------------------------------------- /sgm/modules/diffusionmodules/denoiser_scaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/modules/diffusionmodules/denoiser_scaling.py -------------------------------------------------------------------------------- /sgm/modules/diffusionmodules/denoiser_weighting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/modules/diffusionmodules/denoiser_weighting.py -------------------------------------------------------------------------------- /sgm/modules/diffusionmodules/discretizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/modules/diffusionmodules/discretizer.py -------------------------------------------------------------------------------- /sgm/modules/diffusionmodules/guiders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/modules/diffusionmodules/guiders.py -------------------------------------------------------------------------------- /sgm/modules/diffusionmodules/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/modules/diffusionmodules/loss.py -------------------------------------------------------------------------------- /sgm/modules/diffusionmodules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/modules/diffusionmodules/model.py -------------------------------------------------------------------------------- /sgm/modules/diffusionmodules/openaimodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/modules/diffusionmodules/openaimodel.py -------------------------------------------------------------------------------- /sgm/modules/diffusionmodules/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/modules/diffusionmodules/sampling.py -------------------------------------------------------------------------------- /sgm/modules/diffusionmodules/sampling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/modules/diffusionmodules/sampling_utils.py -------------------------------------------------------------------------------- /sgm/modules/diffusionmodules/sigma_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/modules/diffusionmodules/sigma_sampling.py -------------------------------------------------------------------------------- /sgm/modules/diffusionmodules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/modules/diffusionmodules/util.py -------------------------------------------------------------------------------- /sgm/modules/diffusionmodules/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/modules/diffusionmodules/wrappers.py -------------------------------------------------------------------------------- /sgm/modules/distributions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sgm/modules/distributions/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/modules/distributions/distributions.py -------------------------------------------------------------------------------- /sgm/modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/modules/ema.py -------------------------------------------------------------------------------- /sgm/modules/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sgm/modules/encoders/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/modules/encoders/modules.py -------------------------------------------------------------------------------- /sgm/modules/predictors/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/modules/predictors/model.py -------------------------------------------------------------------------------- /sgm/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/sgm/util.py -------------------------------------------------------------------------------- /src/parseq/.github/contexts-example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/.github/contexts-example.png -------------------------------------------------------------------------------- /src/parseq/.github/gh-teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/.github/gh-teaser.png -------------------------------------------------------------------------------- /src/parseq/.github/system.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/.github/system.png -------------------------------------------------------------------------------- /src/parseq/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/.gitignore -------------------------------------------------------------------------------- /src/parseq/Datasets.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/Datasets.md -------------------------------------------------------------------------------- /src/parseq/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/LICENSE -------------------------------------------------------------------------------- /src/parseq/NOTICE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/NOTICE -------------------------------------------------------------------------------- /src/parseq/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/README.md -------------------------------------------------------------------------------- /src/parseq/bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/bench.py -------------------------------------------------------------------------------- /src/parseq/configs/bench.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/configs/bench.yaml -------------------------------------------------------------------------------- /src/parseq/configs/charset/36_lowercase.yaml: -------------------------------------------------------------------------------- 1 | # @package _global_ 2 | model: 3 | charset_train: "0123456789abcdefghijklmnopqrstuvwxyz" 4 | -------------------------------------------------------------------------------- /src/parseq/configs/charset/62_mixed-case.yaml: -------------------------------------------------------------------------------- 1 | # @package _global_ 2 | model: 3 | charset_train: "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ" 4 | -------------------------------------------------------------------------------- /src/parseq/configs/charset/94_full.yaml: -------------------------------------------------------------------------------- 1 | # @package _global_ 2 | model: 3 | charset_train: "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ!\"#$%&'()*+,-./:;<=>?@[\\]^_`{|}~" 4 | -------------------------------------------------------------------------------- /src/parseq/configs/dataset/real.yaml: -------------------------------------------------------------------------------- 1 | # @package _global_ 2 | data: 3 | train_dir: real 4 | -------------------------------------------------------------------------------- /src/parseq/configs/dataset/synth.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/configs/dataset/synth.yaml -------------------------------------------------------------------------------- /src/parseq/configs/experiment/abinet-sv.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/configs/experiment/abinet-sv.yaml -------------------------------------------------------------------------------- /src/parseq/configs/experiment/abinet.yaml: -------------------------------------------------------------------------------- 1 | # @package _global_ 2 | defaults: 3 | - override /model: abinet 4 | -------------------------------------------------------------------------------- /src/parseq/configs/experiment/crnn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/configs/experiment/crnn.yaml -------------------------------------------------------------------------------- /src/parseq/configs/experiment/parseq-patch16-224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/configs/experiment/parseq-patch16-224.yaml -------------------------------------------------------------------------------- /src/parseq/configs/experiment/parseq-tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/configs/experiment/parseq-tiny.yaml -------------------------------------------------------------------------------- /src/parseq/configs/experiment/parseq.yaml: -------------------------------------------------------------------------------- 1 | # @package _global_ 2 | defaults: 3 | - override /model: parseq 4 | -------------------------------------------------------------------------------- /src/parseq/configs/experiment/trba.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/configs/experiment/trba.yaml -------------------------------------------------------------------------------- /src/parseq/configs/experiment/trbc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/configs/experiment/trbc.yaml -------------------------------------------------------------------------------- /src/parseq/configs/experiment/tune_abinet-lm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/configs/experiment/tune_abinet-lm.yaml -------------------------------------------------------------------------------- /src/parseq/configs/experiment/vitstr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/configs/experiment/vitstr.yaml -------------------------------------------------------------------------------- /src/parseq/configs/main.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/configs/main.yaml -------------------------------------------------------------------------------- /src/parseq/configs/model/abinet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/configs/model/abinet.yaml -------------------------------------------------------------------------------- /src/parseq/configs/model/crnn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/configs/model/crnn.yaml -------------------------------------------------------------------------------- /src/parseq/configs/model/parseq.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/configs/model/parseq.yaml -------------------------------------------------------------------------------- /src/parseq/configs/model/trba.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/configs/model/trba.yaml -------------------------------------------------------------------------------- /src/parseq/configs/model/vitstr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/configs/model/vitstr.yaml -------------------------------------------------------------------------------- /src/parseq/configs/tune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/configs/tune.yaml -------------------------------------------------------------------------------- /src/parseq/demo_images/art-01107.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/demo_images/art-01107.jpg -------------------------------------------------------------------------------- /src/parseq/demo_images/coco-1166773.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/demo_images/coco-1166773.jpg -------------------------------------------------------------------------------- /src/parseq/demo_images/cute-184.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/demo_images/cute-184.jpg -------------------------------------------------------------------------------- /src/parseq/demo_images/ic13_word_256.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/demo_images/ic13_word_256.png -------------------------------------------------------------------------------- /src/parseq/demo_images/ic15_word_26.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/demo_images/ic15_word_26.png -------------------------------------------------------------------------------- /src/parseq/demo_images/uber-27491.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/demo_images/uber-27491.jpg -------------------------------------------------------------------------------- /src/parseq/hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/hubconf.py -------------------------------------------------------------------------------- /src/parseq/read.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/read.py -------------------------------------------------------------------------------- /src/parseq/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/requirements.txt -------------------------------------------------------------------------------- /src/parseq/setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/setup.cfg -------------------------------------------------------------------------------- /src/parseq/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/setup.py -------------------------------------------------------------------------------- /src/parseq/strhub/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/parseq/strhub/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/parseq/strhub/data/aa_overrides.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/data/aa_overrides.py -------------------------------------------------------------------------------- /src/parseq/strhub/data/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/data/augment.py -------------------------------------------------------------------------------- /src/parseq/strhub/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/data/dataset.py -------------------------------------------------------------------------------- /src/parseq/strhub/data/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/data/module.py -------------------------------------------------------------------------------- /src/parseq/strhub/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/data/utils.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/parseq/strhub/models/abinet/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/abinet/LICENSE -------------------------------------------------------------------------------- /src/parseq/strhub/models/abinet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/abinet/__init__.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/abinet/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/abinet/attention.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/abinet/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/abinet/backbone.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/abinet/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/abinet/model.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/abinet/model_abinet_iter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/abinet/model_abinet_iter.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/abinet/model_alignment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/abinet/model_alignment.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/abinet/model_language.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/abinet/model_language.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/abinet/model_vision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/abinet/model_vision.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/abinet/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/abinet/resnet.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/abinet/system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/abinet/system.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/abinet/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/abinet/transformer.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/base.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/crnn/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/crnn/LICENSE -------------------------------------------------------------------------------- /src/parseq/strhub/models/crnn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/crnn/__init__.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/crnn/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/crnn/model.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/crnn/system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/crnn/system.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/modules.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/parseq/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/parseq/strhub/models/parseq/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/parseq/modules.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/parseq/system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/parseq/system.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/trba/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/trba/__init__.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/trba/feature_extraction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/trba/feature_extraction.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/trba/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/trba/model.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/trba/prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/trba/prediction.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/trba/system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/trba/system.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/trba/transformation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/trba/transformation.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/utils.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/vitstr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/vitstr/__init__.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/vitstr/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/vitstr/model.py -------------------------------------------------------------------------------- /src/parseq/strhub/models/vitstr/system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/strhub/models/vitstr/system.py -------------------------------------------------------------------------------- /src/parseq/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/test.py -------------------------------------------------------------------------------- /src/parseq/tools/art_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/tools/art_converter.py -------------------------------------------------------------------------------- /src/parseq/tools/case_sensitive_str_datasets_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/tools/case_sensitive_str_datasets_converter.py -------------------------------------------------------------------------------- /src/parseq/tools/coco_2_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/tools/coco_2_converter.py -------------------------------------------------------------------------------- /src/parseq/tools/coco_text_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/tools/coco_text_converter.py -------------------------------------------------------------------------------- /src/parseq/tools/create_lmdb_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/tools/create_lmdb_dataset.py -------------------------------------------------------------------------------- /src/parseq/tools/filter_lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/tools/filter_lmdb.py -------------------------------------------------------------------------------- /src/parseq/tools/lsvt_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/tools/lsvt_converter.py -------------------------------------------------------------------------------- /src/parseq/tools/mlt19_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/tools/mlt19_converter.py -------------------------------------------------------------------------------- /src/parseq/tools/openvino_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/tools/openvino_converter.py -------------------------------------------------------------------------------- /src/parseq/tools/test_abinet_lm_acc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/tools/test_abinet_lm_acc.py -------------------------------------------------------------------------------- /src/parseq/tools/textocr_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/tools/textocr_converter.py -------------------------------------------------------------------------------- /src/parseq/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/train.py -------------------------------------------------------------------------------- /src/parseq/tune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/src/parseq/tune.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/train.py -------------------------------------------------------------------------------- /util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZYM-PKU/UDiffText/HEAD/util.py --------------------------------------------------------------------------------