├── .gitignore ├── LICENSE ├── README.md ├── Retrieval.py ├── configs ├── PS_cuhk_pedes.yaml ├── PS_icfg_pedes.yaml ├── PS_rstp_reid.yaml └── config_bert.json ├── data_process.py ├── dataset ├── __init__.py ├── ps_dataset.py └── utils.py ├── images └── architecture.jpg ├── models ├── __init__.py ├── model_person_search.py ├── tokenization_bert.py ├── vit.py └── xbert.py ├── optim ├── __init__.py ├── adafactor.py ├── adahessian.py ├── adamp.py ├── adamw.py ├── lookahead.py ├── nadam.py ├── novograd.py ├── nvnovograd.py ├── optim_factory.py ├── radam.py ├── rmsprop_tf.py └── sgdp.py ├── scheduler ├── __init__.py ├── cosine_lr.py ├── plateau_lr.py ├── scheduler.py ├── scheduler_factory.py ├── step_lr.py └── tanh_lr.py ├── shell ├── cuhk-eval.sh ├── cuhk-train.sh ├── data_process.sh ├── icfg-eval.sh ├── icfg-train.sh ├── rstp-eval.sh └── rstp-train.sh └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/README.md -------------------------------------------------------------------------------- /Retrieval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/Retrieval.py -------------------------------------------------------------------------------- /configs/PS_cuhk_pedes.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/configs/PS_cuhk_pedes.yaml -------------------------------------------------------------------------------- /configs/PS_icfg_pedes.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/configs/PS_icfg_pedes.yaml -------------------------------------------------------------------------------- /configs/PS_rstp_reid.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/configs/PS_rstp_reid.yaml -------------------------------------------------------------------------------- /configs/config_bert.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/configs/config_bert.json -------------------------------------------------------------------------------- /data_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/data_process.py -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/dataset/__init__.py -------------------------------------------------------------------------------- /dataset/ps_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/dataset/ps_dataset.py -------------------------------------------------------------------------------- /dataset/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/dataset/utils.py -------------------------------------------------------------------------------- /images/architecture.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/images/architecture.jpg -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /models/model_person_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/models/model_person_search.py -------------------------------------------------------------------------------- /models/tokenization_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/models/tokenization_bert.py -------------------------------------------------------------------------------- /models/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/models/vit.py -------------------------------------------------------------------------------- /models/xbert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/models/xbert.py -------------------------------------------------------------------------------- /optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/optim/__init__.py -------------------------------------------------------------------------------- /optim/adafactor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/optim/adafactor.py -------------------------------------------------------------------------------- /optim/adahessian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/optim/adahessian.py -------------------------------------------------------------------------------- /optim/adamp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/optim/adamp.py -------------------------------------------------------------------------------- /optim/adamw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/optim/adamw.py -------------------------------------------------------------------------------- /optim/lookahead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/optim/lookahead.py -------------------------------------------------------------------------------- /optim/nadam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/optim/nadam.py -------------------------------------------------------------------------------- /optim/novograd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/optim/novograd.py -------------------------------------------------------------------------------- /optim/nvnovograd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/optim/nvnovograd.py -------------------------------------------------------------------------------- /optim/optim_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/optim/optim_factory.py -------------------------------------------------------------------------------- /optim/radam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/optim/radam.py -------------------------------------------------------------------------------- /optim/rmsprop_tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/optim/rmsprop_tf.py -------------------------------------------------------------------------------- /optim/sgdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/optim/sgdp.py -------------------------------------------------------------------------------- /scheduler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/scheduler/__init__.py -------------------------------------------------------------------------------- /scheduler/cosine_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/scheduler/cosine_lr.py -------------------------------------------------------------------------------- /scheduler/plateau_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/scheduler/plateau_lr.py -------------------------------------------------------------------------------- /scheduler/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/scheduler/scheduler.py -------------------------------------------------------------------------------- /scheduler/scheduler_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/scheduler/scheduler_factory.py -------------------------------------------------------------------------------- /scheduler/step_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/scheduler/step_lr.py -------------------------------------------------------------------------------- /scheduler/tanh_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/scheduler/tanh_lr.py -------------------------------------------------------------------------------- /shell/cuhk-eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/shell/cuhk-eval.sh -------------------------------------------------------------------------------- /shell/cuhk-train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/shell/cuhk-train.sh -------------------------------------------------------------------------------- /shell/data_process.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/shell/data_process.sh -------------------------------------------------------------------------------- /shell/icfg-eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/shell/icfg-eval.sh -------------------------------------------------------------------------------- /shell/icfg-train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/shell/icfg-train.sh -------------------------------------------------------------------------------- /shell/rstp-eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/shell/rstp-eval.sh -------------------------------------------------------------------------------- /shell/rstp-train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/shell/rstp-train.sh -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Flame-Chasers/RaSa/HEAD/utils.py --------------------------------------------------------------------------------