├── LICENSE ├── README.md ├── Retrieval.py ├── assets ├── badsamples.jpg ├── chart1.jpg ├── cuhk_freq50.jpg ├── examples.jpg ├── framework.jpg ├── mals_fre50.jpg ├── mals_rst_30.jpg ├── readme.txt └── result.jpg ├── configs ├── Retrieval_cuhk.yaml ├── Retrieval_gene.yaml ├── Retrieval_icfg.yaml ├── Retrieval_pa100k.yaml ├── Retrieval_rstp.yaml ├── config_bert.json └── config_swinB_384.json ├── dataset ├── __init__.py ├── eda.py ├── randaugment.py ├── random_erasing.py ├── re_dataset.py └── utils.py ├── models ├── __init__.py ├── aptm.py ├── bert.py ├── model_retrieval.py ├── swin_transformer.py └── tokenization_bert.py ├── optim.py ├── reTools.py ├── requirements.txt ├── run.py ├── scheduler.py ├── train_pa100ks.py ├── train_tools.py ├── trains.py └── utils └── __init__.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/README.md -------------------------------------------------------------------------------- /Retrieval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/Retrieval.py -------------------------------------------------------------------------------- /assets/badsamples.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/assets/badsamples.jpg -------------------------------------------------------------------------------- /assets/chart1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/assets/chart1.jpg -------------------------------------------------------------------------------- /assets/cuhk_freq50.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/assets/cuhk_freq50.jpg -------------------------------------------------------------------------------- /assets/examples.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/assets/examples.jpg -------------------------------------------------------------------------------- /assets/framework.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/assets/framework.jpg -------------------------------------------------------------------------------- /assets/mals_fre50.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/assets/mals_fre50.jpg -------------------------------------------------------------------------------- /assets/mals_rst_30.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/assets/mals_rst_30.jpg -------------------------------------------------------------------------------- /assets/readme.txt: -------------------------------------------------------------------------------- 1 | assets 2 | -------------------------------------------------------------------------------- /assets/result.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/assets/result.jpg -------------------------------------------------------------------------------- /configs/Retrieval_cuhk.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/configs/Retrieval_cuhk.yaml -------------------------------------------------------------------------------- /configs/Retrieval_gene.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/configs/Retrieval_gene.yaml -------------------------------------------------------------------------------- /configs/Retrieval_icfg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/configs/Retrieval_icfg.yaml -------------------------------------------------------------------------------- /configs/Retrieval_pa100k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/configs/Retrieval_pa100k.yaml -------------------------------------------------------------------------------- /configs/Retrieval_rstp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/configs/Retrieval_rstp.yaml -------------------------------------------------------------------------------- /configs/config_bert.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/configs/config_bert.json -------------------------------------------------------------------------------- /configs/config_swinB_384.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/configs/config_swinB_384.json -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/dataset/__init__.py -------------------------------------------------------------------------------- /dataset/eda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/dataset/eda.py -------------------------------------------------------------------------------- /dataset/randaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/dataset/randaugment.py -------------------------------------------------------------------------------- /dataset/random_erasing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/dataset/random_erasing.py -------------------------------------------------------------------------------- /dataset/re_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/dataset/re_dataset.py -------------------------------------------------------------------------------- /dataset/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/dataset/utils.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/aptm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/models/aptm.py -------------------------------------------------------------------------------- /models/bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/models/bert.py -------------------------------------------------------------------------------- /models/model_retrieval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/models/model_retrieval.py -------------------------------------------------------------------------------- /models/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/models/swin_transformer.py -------------------------------------------------------------------------------- /models/tokenization_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/models/tokenization_bert.py -------------------------------------------------------------------------------- /optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/optim.py -------------------------------------------------------------------------------- /reTools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/reTools.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/run.py -------------------------------------------------------------------------------- /scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/scheduler.py -------------------------------------------------------------------------------- /train_pa100ks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/train_pa100ks.py -------------------------------------------------------------------------------- /train_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/train_tools.py -------------------------------------------------------------------------------- /trains.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/trains.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuyu-XJTU/APTM/HEAD/utils/__init__.py --------------------------------------------------------------------------------