├── .gitignore ├── README.md ├── data_configs ├── eval │ ├── scut-enstext.yaml │ ├── tampered-ic13.yaml │ └── textseg.yaml └── train │ ├── scut-enstext.yaml │ ├── tampered-ic13.yaml │ └── textseg.yaml ├── datasets ├── __init__.py ├── collate_fn.py ├── multitask_dataloader.py ├── tampered_text_detection │ ├── __init__.py │ ├── tampered_ic13.py │ └── transforms.py ├── text_removal │ ├── __init__.py │ ├── scut_enstext.py │ └── transforms.py └── text_segmentation │ ├── __init__.py │ ├── textseg.py │ └── transforms.py ├── engine ├── __init__.py ├── train.py └── val.py ├── eval ├── segmentation.py └── text_removal │ ├── __init__.py │ ├── dataloader.py │ ├── evaluation.py │ └── gauss.py ├── figures └── method.svg ├── main.py ├── models ├── __init__.py ├── block.py ├── decoder │ ├── __init__.py │ └── swinv2_decoder.py ├── encoder │ ├── __init__.py │ ├── swin_transformer_v2.py │ └── swinv2_encoder.py ├── task_prompt.py ├── upocr.py └── vgg16.py ├── optim ├── __init__.py ├── criterion │ ├── tampered_text_detection.py │ ├── text_removal.py │ └── text_segmentation.py └── lr_scheduler.py ├── requirements.txt └── utils ├── __init__.py ├── checkpointer.py ├── dist.py ├── logger.py ├── misc.py ├── parser.py └── visualize.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/README.md -------------------------------------------------------------------------------- /data_configs/eval/scut-enstext.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/data_configs/eval/scut-enstext.yaml -------------------------------------------------------------------------------- /data_configs/eval/tampered-ic13.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/data_configs/eval/tampered-ic13.yaml -------------------------------------------------------------------------------- /data_configs/eval/textseg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/data_configs/eval/textseg.yaml -------------------------------------------------------------------------------- /data_configs/train/scut-enstext.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/data_configs/train/scut-enstext.yaml -------------------------------------------------------------------------------- /data_configs/train/tampered-ic13.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/data_configs/train/tampered-ic13.yaml -------------------------------------------------------------------------------- /data_configs/train/textseg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/data_configs/train/textseg.yaml -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/collate_fn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/datasets/collate_fn.py -------------------------------------------------------------------------------- /datasets/multitask_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/datasets/multitask_dataloader.py -------------------------------------------------------------------------------- /datasets/tampered_text_detection/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/datasets/tampered_text_detection/__init__.py -------------------------------------------------------------------------------- /datasets/tampered_text_detection/tampered_ic13.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/datasets/tampered_text_detection/tampered_ic13.py -------------------------------------------------------------------------------- /datasets/tampered_text_detection/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/datasets/tampered_text_detection/transforms.py -------------------------------------------------------------------------------- /datasets/text_removal/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/datasets/text_removal/__init__.py -------------------------------------------------------------------------------- /datasets/text_removal/scut_enstext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/datasets/text_removal/scut_enstext.py -------------------------------------------------------------------------------- /datasets/text_removal/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/datasets/text_removal/transforms.py -------------------------------------------------------------------------------- /datasets/text_segmentation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/datasets/text_segmentation/__init__.py -------------------------------------------------------------------------------- /datasets/text_segmentation/textseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/datasets/text_segmentation/textseg.py -------------------------------------------------------------------------------- /datasets/text_segmentation/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/datasets/text_segmentation/transforms.py -------------------------------------------------------------------------------- /engine/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /engine/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/engine/train.py -------------------------------------------------------------------------------- /engine/val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/engine/val.py -------------------------------------------------------------------------------- /eval/segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/eval/segmentation.py -------------------------------------------------------------------------------- /eval/text_removal/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /eval/text_removal/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/eval/text_removal/dataloader.py -------------------------------------------------------------------------------- /eval/text_removal/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/eval/text_removal/evaluation.py -------------------------------------------------------------------------------- /eval/text_removal/gauss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/eval/text_removal/gauss.py -------------------------------------------------------------------------------- /figures/method.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/figures/method.svg -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/main.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/models/block.py -------------------------------------------------------------------------------- /models/decoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/models/decoder/__init__.py -------------------------------------------------------------------------------- /models/decoder/swinv2_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/models/decoder/swinv2_decoder.py -------------------------------------------------------------------------------- /models/encoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/models/encoder/__init__.py -------------------------------------------------------------------------------- /models/encoder/swin_transformer_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/models/encoder/swin_transformer_v2.py -------------------------------------------------------------------------------- /models/encoder/swinv2_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/models/encoder/swinv2_encoder.py -------------------------------------------------------------------------------- /models/task_prompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/models/task_prompt.py -------------------------------------------------------------------------------- /models/upocr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/models/upocr.py -------------------------------------------------------------------------------- /models/vgg16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/models/vgg16.py -------------------------------------------------------------------------------- /optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/optim/__init__.py -------------------------------------------------------------------------------- /optim/criterion/tampered_text_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/optim/criterion/tampered_text_detection.py -------------------------------------------------------------------------------- /optim/criterion/text_removal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/optim/criterion/text_removal.py -------------------------------------------------------------------------------- /optim/criterion/text_segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/optim/criterion/text_segmentation.py -------------------------------------------------------------------------------- /optim/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/optim/lr_scheduler.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/requirements.txt -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved 2 | -------------------------------------------------------------------------------- /utils/checkpointer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/utils/checkpointer.py -------------------------------------------------------------------------------- /utils/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/utils/dist.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/utils/misc.py -------------------------------------------------------------------------------- /utils/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/utils/parser.py -------------------------------------------------------------------------------- /utils/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/UPOCR/HEAD/utils/visualize.py --------------------------------------------------------------------------------