├── Figures ├── Comparison.png ├── Result1.png ├── Result2.png └── Structure.png ├── README.md ├── arun_log └── job-13005292-20211112152156.log ├── config ├── __init__.py └── defaults.py ├── configs ├── OCC_Duke │ └── vit_base.yml └── transformer_base.yml ├── datasets ├── __init__.py ├── bases.py ├── dukemtmcreid.py ├── keypoint_test.txt ├── keypoint_train.txt ├── make_dataloader.py ├── market1501.py ├── msmt17.py ├── occ_duke.py ├── preprocessing.py ├── sampler.py ├── sampler_ddp.py ├── vehicleid.py └── veri.py ├── loss ├── __init__.py ├── arcface.py ├── center_loss.py ├── infonce.py ├── make_loss.py ├── metric_learning.py ├── softmax_loss.py └── triplet_loss.py ├── model ├── __init__.py ├── backbones │ ├── __init__.py │ ├── resnet.py │ └── vit_pytorch.py ├── diversity_model.py ├── hm.py ├── make_model.py └── perceiver.py ├── processor ├── __init__.py └── processor.py ├── requirements.txt ├── solver ├── __init__.py ├── cosine_lr.py ├── lr_scheduler.py ├── make_optimizer.py ├── scheduler.py └── scheduler_factory.py ├── test.py ├── test_individual.py ├── train.py └── utils ├── __init__.py ├── iotools.py ├── logger.py ├── meter.py ├── metrics.py └── reranking.py /Figures/Comparison.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/Figures/Comparison.png -------------------------------------------------------------------------------- /Figures/Result1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/Figures/Result1.png -------------------------------------------------------------------------------- /Figures/Result2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/Figures/Result2.png -------------------------------------------------------------------------------- /Figures/Structure.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/Figures/Structure.png -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/README.md -------------------------------------------------------------------------------- /arun_log/job-13005292-20211112152156.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/arun_log/job-13005292-20211112152156.log -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/config/__init__.py -------------------------------------------------------------------------------- /config/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/config/defaults.py -------------------------------------------------------------------------------- /configs/OCC_Duke/vit_base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/configs/OCC_Duke/vit_base.yml -------------------------------------------------------------------------------- /configs/transformer_base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/configs/transformer_base.yml -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/bases.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/datasets/bases.py -------------------------------------------------------------------------------- /datasets/dukemtmcreid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/datasets/dukemtmcreid.py -------------------------------------------------------------------------------- /datasets/keypoint_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/datasets/keypoint_test.txt -------------------------------------------------------------------------------- /datasets/keypoint_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/datasets/keypoint_train.txt -------------------------------------------------------------------------------- /datasets/make_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/datasets/make_dataloader.py -------------------------------------------------------------------------------- /datasets/market1501.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/datasets/market1501.py -------------------------------------------------------------------------------- /datasets/msmt17.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/datasets/msmt17.py -------------------------------------------------------------------------------- /datasets/occ_duke.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/datasets/occ_duke.py -------------------------------------------------------------------------------- /datasets/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/datasets/preprocessing.py -------------------------------------------------------------------------------- /datasets/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/datasets/sampler.py -------------------------------------------------------------------------------- /datasets/sampler_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/datasets/sampler_ddp.py -------------------------------------------------------------------------------- /datasets/vehicleid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/datasets/vehicleid.py -------------------------------------------------------------------------------- /datasets/veri.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/datasets/veri.py -------------------------------------------------------------------------------- /loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/loss/__init__.py -------------------------------------------------------------------------------- /loss/arcface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/loss/arcface.py -------------------------------------------------------------------------------- /loss/center_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/loss/center_loss.py -------------------------------------------------------------------------------- /loss/infonce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/loss/infonce.py -------------------------------------------------------------------------------- /loss/make_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/loss/make_loss.py -------------------------------------------------------------------------------- /loss/metric_learning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/loss/metric_learning.py -------------------------------------------------------------------------------- /loss/softmax_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/loss/softmax_loss.py -------------------------------------------------------------------------------- /loss/triplet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/loss/triplet_loss.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/backbones/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/model/backbones/resnet.py -------------------------------------------------------------------------------- /model/backbones/vit_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/model/backbones/vit_pytorch.py -------------------------------------------------------------------------------- /model/diversity_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/model/diversity_model.py -------------------------------------------------------------------------------- /model/hm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/model/hm.py -------------------------------------------------------------------------------- /model/make_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/model/make_model.py -------------------------------------------------------------------------------- /model/perceiver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/model/perceiver.py -------------------------------------------------------------------------------- /processor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/processor/__init__.py -------------------------------------------------------------------------------- /processor/processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/processor/processor.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/requirements.txt -------------------------------------------------------------------------------- /solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/solver/__init__.py -------------------------------------------------------------------------------- /solver/cosine_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/solver/cosine_lr.py -------------------------------------------------------------------------------- /solver/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/solver/lr_scheduler.py -------------------------------------------------------------------------------- /solver/make_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/solver/make_optimizer.py -------------------------------------------------------------------------------- /solver/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/solver/scheduler.py -------------------------------------------------------------------------------- /solver/scheduler_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/solver/scheduler_factory.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/test.py -------------------------------------------------------------------------------- /test_individual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/test_individual.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/iotools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/utils/iotools.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/utils/meter.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/reranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZacharyWang-007/FED-Occluded-ReID/HEAD/utils/reranking.py --------------------------------------------------------------------------------