├── README.md ├── config ├── __init__.py └── defaults.py ├── configs ├── softmax_triplet.yaml └── softmax_triplet_pd.yaml ├── data ├── __init__.py ├── build.py ├── collate_batch.py ├── datasets │ ├── VeRi.py │ ├── VeRi_wild.py │ ├── VehicleID.py │ ├── __init__.py │ ├── bases.py │ └── dataset_loader.py ├── samplers │ ├── __init__.py │ ├── sampler_ddp.py │ └── triplet_sampler.py └── transforms │ ├── __init__.py │ ├── build.py │ └── transformer.py ├── experiments ├── train_res50_UFDN_776.sh ├── train_res50_UFDN_VehicleID.sh ├── train_swin_UFDN_776.sh └── train_swin_UFDN_VehicleID.sh ├── figs ├── overview.jpg └── overview.pdf ├── layers ├── __init__.py ├── focal_loss.py ├── ns_structure_loss.py ├── triplet_loss.py └── xent_loss.py ├── modeling ├── __init__.py ├── backbone │ ├── PDecouple_resnet.py │ ├── PDecouple_swin.py │ ├── __init__.py │ ├── resnet.py │ ├── resnet_sync.py │ └── vit.py ├── baseline.py └── baseline_pd.py ├── requirements.txt ├── solver ├── __init__.py ├── build.py ├── cosine_lr.py ├── lr_scheduler.py ├── ranger.py ├── scheduler.py └── scheduler_factory.py ├── test.sh ├── tools ├── dist_train.py └── train │ ├── AverageMeter.py │ ├── __init__.py │ ├── train_normal.py │ └── train_normal_pd.py └── utils ├── __init__.py ├── analyse.py ├── iotools.py ├── logger.py ├── re_ranking.py └── reid_metric.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/README.md -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/config/__init__.py -------------------------------------------------------------------------------- /config/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/config/defaults.py -------------------------------------------------------------------------------- /configs/softmax_triplet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/configs/softmax_triplet.yaml -------------------------------------------------------------------------------- /configs/softmax_triplet_pd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/configs/softmax_triplet_pd.yaml -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/data/build.py -------------------------------------------------------------------------------- /data/collate_batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/data/collate_batch.py -------------------------------------------------------------------------------- /data/datasets/VeRi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/data/datasets/VeRi.py -------------------------------------------------------------------------------- /data/datasets/VeRi_wild.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/data/datasets/VeRi_wild.py -------------------------------------------------------------------------------- /data/datasets/VehicleID.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/data/datasets/VehicleID.py -------------------------------------------------------------------------------- /data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/data/datasets/__init__.py -------------------------------------------------------------------------------- /data/datasets/bases.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/data/datasets/bases.py -------------------------------------------------------------------------------- /data/datasets/dataset_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/data/datasets/dataset_loader.py -------------------------------------------------------------------------------- /data/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/data/samplers/__init__.py -------------------------------------------------------------------------------- /data/samplers/sampler_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/data/samplers/sampler_ddp.py -------------------------------------------------------------------------------- /data/samplers/triplet_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/data/samplers/triplet_sampler.py -------------------------------------------------------------------------------- /data/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/data/transforms/__init__.py -------------------------------------------------------------------------------- /data/transforms/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/data/transforms/build.py -------------------------------------------------------------------------------- /data/transforms/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/data/transforms/transformer.py -------------------------------------------------------------------------------- /experiments/train_res50_UFDN_776.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/experiments/train_res50_UFDN_776.sh -------------------------------------------------------------------------------- /experiments/train_res50_UFDN_VehicleID.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/experiments/train_res50_UFDN_VehicleID.sh -------------------------------------------------------------------------------- /experiments/train_swin_UFDN_776.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/experiments/train_swin_UFDN_776.sh -------------------------------------------------------------------------------- /experiments/train_swin_UFDN_VehicleID.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/experiments/train_swin_UFDN_VehicleID.sh -------------------------------------------------------------------------------- /figs/overview.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/figs/overview.jpg -------------------------------------------------------------------------------- /figs/overview.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/figs/overview.pdf -------------------------------------------------------------------------------- /layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/layers/__init__.py -------------------------------------------------------------------------------- /layers/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/layers/focal_loss.py -------------------------------------------------------------------------------- /layers/ns_structure_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/layers/ns_structure_loss.py -------------------------------------------------------------------------------- /layers/triplet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/layers/triplet_loss.py -------------------------------------------------------------------------------- /layers/xent_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/layers/xent_loss.py -------------------------------------------------------------------------------- /modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/modeling/__init__.py -------------------------------------------------------------------------------- /modeling/backbone/PDecouple_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/modeling/backbone/PDecouple_resnet.py -------------------------------------------------------------------------------- /modeling/backbone/PDecouple_swin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/modeling/backbone/PDecouple_swin.py -------------------------------------------------------------------------------- /modeling/backbone/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modeling/backbone/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/modeling/backbone/resnet.py -------------------------------------------------------------------------------- /modeling/backbone/resnet_sync.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/modeling/backbone/resnet_sync.py -------------------------------------------------------------------------------- /modeling/backbone/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/modeling/backbone/vit.py -------------------------------------------------------------------------------- /modeling/baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/modeling/baseline.py -------------------------------------------------------------------------------- /modeling/baseline_pd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/modeling/baseline_pd.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/requirements.txt -------------------------------------------------------------------------------- /solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/solver/__init__.py -------------------------------------------------------------------------------- /solver/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/solver/build.py -------------------------------------------------------------------------------- /solver/cosine_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/solver/cosine_lr.py -------------------------------------------------------------------------------- /solver/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/solver/lr_scheduler.py -------------------------------------------------------------------------------- /solver/ranger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/solver/ranger.py -------------------------------------------------------------------------------- /solver/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/solver/scheduler.py -------------------------------------------------------------------------------- /solver/scheduler_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/solver/scheduler_factory.py -------------------------------------------------------------------------------- /test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/test.sh -------------------------------------------------------------------------------- /tools/dist_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/tools/dist_train.py -------------------------------------------------------------------------------- /tools/train/AverageMeter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/tools/train/AverageMeter.py -------------------------------------------------------------------------------- /tools/train/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/tools/train/__init__.py -------------------------------------------------------------------------------- /tools/train/train_normal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/tools/train/train_normal.py -------------------------------------------------------------------------------- /tools/train/train_normal_pd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/tools/train/train_normal_pd.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/analyse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/utils/analyse.py -------------------------------------------------------------------------------- /utils/iotools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/utils/iotools.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/re_ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/utils/re_ranking.py -------------------------------------------------------------------------------- /utils/reid_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/UFDN-Reid/HEAD/utils/reid_metric.py --------------------------------------------------------------------------------