├── CONTRIBUTION ├── IMG ├── pipeline.png └── test_0000095.jpg ├── LICENSE ├── README.md ├── datasets ├── __init__.py ├── ocr_dataset.py └── sptsv2_transforms.py ├── engine_sptsv2.py ├── eval.py ├── eval_ic15.py ├── main.py ├── models ├── __init__.py ├── backbone.py ├── encoder_decoder.py ├── position_encoding.py └── sptsv2.py ├── predict.py ├── requirements.txt ├── run.sh ├── test.sh └── util ├── __init__.py ├── data.py ├── misc_sptsv2.py └── visualize.py /CONTRIBUTION: -------------------------------------------------------------------------------- 1 | Contributions or pull requests are welcome. -------------------------------------------------------------------------------- /IMG/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/SPTSv2/HEAD/IMG/pipeline.png -------------------------------------------------------------------------------- /IMG/test_0000095.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/SPTSv2/HEAD/IMG/test_0000095.jpg -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/SPTSv2/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/SPTSv2/HEAD/README.md -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/SPTSv2/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/ocr_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/SPTSv2/HEAD/datasets/ocr_dataset.py -------------------------------------------------------------------------------- /datasets/sptsv2_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/SPTSv2/HEAD/datasets/sptsv2_transforms.py -------------------------------------------------------------------------------- /engine_sptsv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/SPTSv2/HEAD/engine_sptsv2.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/SPTSv2/HEAD/eval.py -------------------------------------------------------------------------------- /eval_ic15.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/SPTSv2/HEAD/eval_ic15.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/SPTSv2/HEAD/main.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/SPTSv2/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/SPTSv2/HEAD/models/backbone.py -------------------------------------------------------------------------------- /models/encoder_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/SPTSv2/HEAD/models/encoder_decoder.py -------------------------------------------------------------------------------- /models/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/SPTSv2/HEAD/models/position_encoding.py -------------------------------------------------------------------------------- /models/sptsv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/SPTSv2/HEAD/models/sptsv2.py -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/SPTSv2/HEAD/predict.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/SPTSv2/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/SPTSv2/HEAD/run.sh -------------------------------------------------------------------------------- /test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/SPTSv2/HEAD/test.sh -------------------------------------------------------------------------------- /util/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (2023) Bytedance Ltd. and/or its affiliates -------------------------------------------------------------------------------- /util/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/SPTSv2/HEAD/util/data.py -------------------------------------------------------------------------------- /util/misc_sptsv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/SPTSv2/HEAD/util/misc_sptsv2.py -------------------------------------------------------------------------------- /util/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/SPTSv2/HEAD/util/visualize.py --------------------------------------------------------------------------------