├── .gitignore ├── DATASETS.md ├── LICENSE ├── README.md ├── README_ORIG.md ├── README_Training_and_Testing_Guides.md ├── args.py ├── doc_images ├── Arch.png ├── JET_VIS.png ├── att.png └── qr.png ├── eval_acc.py ├── requirements.txt ├── torchreid ├── __init__.py ├── components │ ├── __init__.py │ ├── attention.py │ ├── branches.py │ ├── dropout.py │ └── shallow_cam.py ├── data_manager.py ├── dataset_loader.py ├── datasets │ ├── __init__.py │ ├── bases.py │ ├── cuhk01.py │ ├── cuhk03.py │ ├── dukemtmcreid.py │ ├── dukemtmcreid_d.py │ ├── dukemtmcvidreid.py │ ├── grid.py │ ├── ilids.py │ ├── ilidsvid.py │ ├── market1501.py │ ├── market1501_d.py │ ├── mars.py │ ├── msmt17.py │ ├── prid2011.py │ ├── prid450s.py │ ├── sensereid.py │ ├── valset.py │ ├── veri.py │ └── viper.py ├── eval_cylib │ ├── Makefile │ ├── __init__.py │ ├── eval_metrics_cy.pyx │ ├── setup.py │ └── test_cython.py ├── eval_metrics.py ├── losses │ ├── __init__.py │ ├── batch_spectral_loss.py │ ├── center_loss.py │ ├── cross_entropy_loss.py │ ├── hard_mine_triplet_loss.py │ ├── incidence_loss.py │ ├── incidence_xent_loss.py │ ├── lowrank_loss.py │ ├── of_penalty.py │ ├── ring_loss.py │ ├── sa_loss.py │ ├── singular_triplet_loss.py │ └── spectral_loss.py ├── models │ ├── __init__.py │ ├── densenet.py │ ├── hacnn.py │ ├── inceptionresnetv2.py │ ├── inceptionv4.py │ ├── mlfn.py │ ├── mobilenetv2.py │ ├── mudeep.py │ ├── nasnet.py │ ├── pcb.py │ ├── resnet.py │ ├── resnetmid.py │ ├── resnext.py │ ├── senet.py │ ├── shufflenet.py │ ├── squeezenet.py │ └── xception.py ├── optimizers.py ├── regularizers │ ├── LSVO.py │ ├── NR.py │ ├── SO.py │ ├── SVDO.py │ ├── SVMO.py │ ├── __init__.py │ └── param_controller.py ├── samplers.py ├── transforms.py └── utils │ ├── __init__.py │ ├── avgmeter.py │ ├── environ.py │ ├── iotools.py │ ├── loggers.py │ ├── nuc_norm.py │ ├── reidtools.py │ └── torchtools.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/.gitignore -------------------------------------------------------------------------------- /DATASETS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/DATASETS.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/README.md -------------------------------------------------------------------------------- /README_ORIG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/README_ORIG.md -------------------------------------------------------------------------------- /README_Training_and_Testing_Guides.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/README_Training_and_Testing_Guides.md -------------------------------------------------------------------------------- /args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/args.py -------------------------------------------------------------------------------- /doc_images/Arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/doc_images/Arch.png -------------------------------------------------------------------------------- /doc_images/JET_VIS.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/doc_images/JET_VIS.png -------------------------------------------------------------------------------- /doc_images/att.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/doc_images/att.png -------------------------------------------------------------------------------- /doc_images/qr.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/doc_images/qr.png -------------------------------------------------------------------------------- /eval_acc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/eval_acc.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/requirements.txt -------------------------------------------------------------------------------- /torchreid/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/__init__.py -------------------------------------------------------------------------------- /torchreid/components/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchreid/components/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/components/attention.py -------------------------------------------------------------------------------- /torchreid/components/branches.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/components/branches.py -------------------------------------------------------------------------------- /torchreid/components/dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/components/dropout.py -------------------------------------------------------------------------------- /torchreid/components/shallow_cam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/components/shallow_cam.py -------------------------------------------------------------------------------- /torchreid/data_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/data_manager.py -------------------------------------------------------------------------------- /torchreid/dataset_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/dataset_loader.py -------------------------------------------------------------------------------- /torchreid/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/datasets/__init__.py -------------------------------------------------------------------------------- /torchreid/datasets/bases.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/datasets/bases.py -------------------------------------------------------------------------------- /torchreid/datasets/cuhk01.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/datasets/cuhk01.py -------------------------------------------------------------------------------- /torchreid/datasets/cuhk03.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/datasets/cuhk03.py -------------------------------------------------------------------------------- /torchreid/datasets/dukemtmcreid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/datasets/dukemtmcreid.py -------------------------------------------------------------------------------- /torchreid/datasets/dukemtmcreid_d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/datasets/dukemtmcreid_d.py -------------------------------------------------------------------------------- /torchreid/datasets/dukemtmcvidreid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/datasets/dukemtmcvidreid.py -------------------------------------------------------------------------------- /torchreid/datasets/grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/datasets/grid.py -------------------------------------------------------------------------------- /torchreid/datasets/ilids.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/datasets/ilids.py -------------------------------------------------------------------------------- /torchreid/datasets/ilidsvid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/datasets/ilidsvid.py -------------------------------------------------------------------------------- /torchreid/datasets/market1501.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/datasets/market1501.py -------------------------------------------------------------------------------- /torchreid/datasets/market1501_d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/datasets/market1501_d.py -------------------------------------------------------------------------------- /torchreid/datasets/mars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/datasets/mars.py -------------------------------------------------------------------------------- /torchreid/datasets/msmt17.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/datasets/msmt17.py -------------------------------------------------------------------------------- /torchreid/datasets/prid2011.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/datasets/prid2011.py -------------------------------------------------------------------------------- /torchreid/datasets/prid450s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/datasets/prid450s.py -------------------------------------------------------------------------------- /torchreid/datasets/sensereid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/datasets/sensereid.py -------------------------------------------------------------------------------- /torchreid/datasets/valset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/datasets/valset.py -------------------------------------------------------------------------------- /torchreid/datasets/veri.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/datasets/veri.py -------------------------------------------------------------------------------- /torchreid/datasets/viper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/datasets/viper.py -------------------------------------------------------------------------------- /torchreid/eval_cylib/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/eval_cylib/Makefile -------------------------------------------------------------------------------- /torchreid/eval_cylib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchreid/eval_cylib/eval_metrics_cy.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/eval_cylib/eval_metrics_cy.pyx -------------------------------------------------------------------------------- /torchreid/eval_cylib/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/eval_cylib/setup.py -------------------------------------------------------------------------------- /torchreid/eval_cylib/test_cython.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/eval_cylib/test_cython.py -------------------------------------------------------------------------------- /torchreid/eval_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/eval_metrics.py -------------------------------------------------------------------------------- /torchreid/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/losses/__init__.py -------------------------------------------------------------------------------- /torchreid/losses/batch_spectral_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/losses/batch_spectral_loss.py -------------------------------------------------------------------------------- /torchreid/losses/center_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/losses/center_loss.py -------------------------------------------------------------------------------- /torchreid/losses/cross_entropy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/losses/cross_entropy_loss.py -------------------------------------------------------------------------------- /torchreid/losses/hard_mine_triplet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/losses/hard_mine_triplet_loss.py -------------------------------------------------------------------------------- /torchreid/losses/incidence_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/losses/incidence_loss.py -------------------------------------------------------------------------------- /torchreid/losses/incidence_xent_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/losses/incidence_xent_loss.py -------------------------------------------------------------------------------- /torchreid/losses/lowrank_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/losses/lowrank_loss.py -------------------------------------------------------------------------------- /torchreid/losses/of_penalty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/losses/of_penalty.py -------------------------------------------------------------------------------- /torchreid/losses/ring_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/losses/ring_loss.py -------------------------------------------------------------------------------- /torchreid/losses/sa_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/losses/sa_loss.py -------------------------------------------------------------------------------- /torchreid/losses/singular_triplet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/losses/singular_triplet_loss.py -------------------------------------------------------------------------------- /torchreid/losses/spectral_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/losses/spectral_loss.py -------------------------------------------------------------------------------- /torchreid/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/models/__init__.py -------------------------------------------------------------------------------- /torchreid/models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/models/densenet.py -------------------------------------------------------------------------------- /torchreid/models/hacnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/models/hacnn.py -------------------------------------------------------------------------------- /torchreid/models/inceptionresnetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/models/inceptionresnetv2.py -------------------------------------------------------------------------------- /torchreid/models/inceptionv4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/models/inceptionv4.py -------------------------------------------------------------------------------- /torchreid/models/mlfn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/models/mlfn.py -------------------------------------------------------------------------------- /torchreid/models/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/models/mobilenetv2.py -------------------------------------------------------------------------------- /torchreid/models/mudeep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/models/mudeep.py -------------------------------------------------------------------------------- /torchreid/models/nasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/models/nasnet.py -------------------------------------------------------------------------------- /torchreid/models/pcb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/models/pcb.py -------------------------------------------------------------------------------- /torchreid/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/models/resnet.py -------------------------------------------------------------------------------- /torchreid/models/resnetmid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/models/resnetmid.py -------------------------------------------------------------------------------- /torchreid/models/resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/models/resnext.py -------------------------------------------------------------------------------- /torchreid/models/senet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/models/senet.py -------------------------------------------------------------------------------- /torchreid/models/shufflenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/models/shufflenet.py -------------------------------------------------------------------------------- /torchreid/models/squeezenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/models/squeezenet.py -------------------------------------------------------------------------------- /torchreid/models/xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/models/xception.py -------------------------------------------------------------------------------- /torchreid/optimizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/optimizers.py -------------------------------------------------------------------------------- /torchreid/regularizers/LSVO.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchreid/regularizers/NR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/regularizers/NR.py -------------------------------------------------------------------------------- /torchreid/regularizers/SO.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/regularizers/SO.py -------------------------------------------------------------------------------- /torchreid/regularizers/SVDO.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/regularizers/SVDO.py -------------------------------------------------------------------------------- /torchreid/regularizers/SVMO.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/regularizers/SVMO.py -------------------------------------------------------------------------------- /torchreid/regularizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/regularizers/__init__.py -------------------------------------------------------------------------------- /torchreid/regularizers/param_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/regularizers/param_controller.py -------------------------------------------------------------------------------- /torchreid/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/samplers.py -------------------------------------------------------------------------------- /torchreid/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/transforms.py -------------------------------------------------------------------------------- /torchreid/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchreid/utils/avgmeter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/utils/avgmeter.py -------------------------------------------------------------------------------- /torchreid/utils/environ.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/utils/environ.py -------------------------------------------------------------------------------- /torchreid/utils/iotools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/utils/iotools.py -------------------------------------------------------------------------------- /torchreid/utils/loggers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/utils/loggers.py -------------------------------------------------------------------------------- /torchreid/utils/nuc_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/utils/nuc_norm.py -------------------------------------------------------------------------------- /torchreid/utils/reidtools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/utils/reidtools.py -------------------------------------------------------------------------------- /torchreid/utils/torchtools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/torchreid/utils/torchtools.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ABD-Net/HEAD/train.py --------------------------------------------------------------------------------