├── README.md ├── cfg.py ├── dataset ├── ParseqAug │ ├── __init__.py │ ├── aa_overrides.py │ └── augment.py ├── __init__.py ├── arial.ttf ├── concatdatasets.py ├── dataset_lmdb.py ├── datasets.py ├── dist_multisrc_batch_sampler.py ├── render_text_mask.py └── transforms.py ├── engine_finetune.py ├── evaluation_metric ├── __init__.py ├── metrics.py └── multi_label_metrics.py ├── fig ├── pipeline.pdf ├── pipline.png └── res.png ├── loss ├── __init__.py ├── balance_cross_entropy_loss.py ├── seqCrossEntropyLoss.py └── seqLabelSmoothingCrossEntropyLoss.py ├── main_finetune.py ├── models ├── __init__.py ├── rec_model.py ├── rec_modules │ ├── __init__.py │ ├── decoder.py │ ├── decoder_utils.py │ ├── fuse_utils.py │ ├── network_utils.py │ ├── stn_head.py │ ├── tps.py │ ├── transformer_layer.py │ └── utils.py └── utils.py ├── optim_factory.py ├── requirements.txt ├── script ├── eva_base.sh ├── eva_nano.sh ├── eva_tiny.sh └── train.sh └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/README.md -------------------------------------------------------------------------------- /cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/cfg.py -------------------------------------------------------------------------------- /dataset/ParseqAug/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/ParseqAug/aa_overrides.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/dataset/ParseqAug/aa_overrides.py -------------------------------------------------------------------------------- /dataset/ParseqAug/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/dataset/ParseqAug/augment.py -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/arial.ttf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/dataset/arial.ttf -------------------------------------------------------------------------------- /dataset/concatdatasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/dataset/concatdatasets.py -------------------------------------------------------------------------------- /dataset/dataset_lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/dataset/dataset_lmdb.py -------------------------------------------------------------------------------- /dataset/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/dataset/datasets.py -------------------------------------------------------------------------------- /dataset/dist_multisrc_batch_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/dataset/dist_multisrc_batch_sampler.py -------------------------------------------------------------------------------- /dataset/render_text_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/dataset/render_text_mask.py -------------------------------------------------------------------------------- /dataset/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/dataset/transforms.py -------------------------------------------------------------------------------- /engine_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/engine_finetune.py -------------------------------------------------------------------------------- /evaluation_metric/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/evaluation_metric/__init__.py -------------------------------------------------------------------------------- /evaluation_metric/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/evaluation_metric/metrics.py -------------------------------------------------------------------------------- /evaluation_metric/multi_label_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/evaluation_metric/multi_label_metrics.py -------------------------------------------------------------------------------- /fig/pipeline.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/fig/pipeline.pdf -------------------------------------------------------------------------------- /fig/pipline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/fig/pipline.png -------------------------------------------------------------------------------- /fig/res.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/fig/res.png -------------------------------------------------------------------------------- /loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/loss/__init__.py -------------------------------------------------------------------------------- /loss/balance_cross_entropy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/loss/balance_cross_entropy_loss.py -------------------------------------------------------------------------------- /loss/seqCrossEntropyLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/loss/seqCrossEntropyLoss.py -------------------------------------------------------------------------------- /loss/seqLabelSmoothingCrossEntropyLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/loss/seqLabelSmoothingCrossEntropyLoss.py -------------------------------------------------------------------------------- /main_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/main_finetune.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/rec_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/models/rec_model.py -------------------------------------------------------------------------------- /models/rec_modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/rec_modules/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/models/rec_modules/decoder.py -------------------------------------------------------------------------------- /models/rec_modules/decoder_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/models/rec_modules/decoder_utils.py -------------------------------------------------------------------------------- /models/rec_modules/fuse_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/models/rec_modules/fuse_utils.py -------------------------------------------------------------------------------- /models/rec_modules/network_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/models/rec_modules/network_utils.py -------------------------------------------------------------------------------- /models/rec_modules/stn_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/models/rec_modules/stn_head.py -------------------------------------------------------------------------------- /models/rec_modules/tps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/models/rec_modules/tps.py -------------------------------------------------------------------------------- /models/rec_modules/transformer_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/models/rec_modules/transformer_layer.py -------------------------------------------------------------------------------- /models/rec_modules/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/models/rec_modules/utils.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/models/utils.py -------------------------------------------------------------------------------- /optim_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/optim_factory.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/requirements.txt -------------------------------------------------------------------------------- /script/eva_base.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/script/eva_base.sh -------------------------------------------------------------------------------- /script/eva_nano.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/script/eva_nano.sh -------------------------------------------------------------------------------- /script/eva_tiny.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/script/eva_tiny.sh -------------------------------------------------------------------------------- /script/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/script/train.sh -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MelosY/CAM/HEAD/utils.py --------------------------------------------------------------------------------