├── .gitignore ├── LICENSE ├── PTGAN.py ├── PTGAN_for_one_test.py ├── README.md ├── code_GAN ├── PTGAN_dataset.py ├── dataset_for_test.py ├── dist.npy ├── gan │ ├── __init__.py │ ├── contrastive_loss.py │ ├── gan_losses.py │ ├── model.py │ ├── networks.py │ ├── options.py │ └── utils │ │ ├── __init__.py │ │ ├── html.py │ │ ├── util.py │ │ └── visualizer.py ├── metrics.py ├── orthogonal_encoder.py ├── processor.py ├── processor_for_test.py ├── reid │ ├── __init__.py │ ├── datasets │ │ ├── __init__.py │ │ └── train.py │ ├── evaluation_metrics │ │ ├── __init__.py │ │ ├── classification.py │ │ └── ranking.py │ ├── evaluators.py │ ├── feature_extraction │ │ ├── __init__.py │ │ ├── cnn.py │ │ └── database.py │ ├── models │ │ ├── __init__.py │ │ ├── baseline.py │ │ ├── embedding.py │ │ ├── multi_branch.py │ │ └── resnet.py │ ├── trainers.py │ └── utils │ │ ├── __init__.py │ │ ├── data │ │ ├── __init__.py │ │ ├── dataset.py │ │ ├── preprocessor.py │ │ ├── sampler.py │ │ └── transforms.py │ │ ├── logging.py │ │ ├── meters.py │ │ ├── osutils.py │ │ └── serialization.py ├── test_processor.py ├── train.py └── weights │ └── GAN_stage_2 │ ├── opt.txt │ └── web │ └── index.html ├── config ├── __init__.py └── defaults.py ├── configs ├── camera_view │ ├── camera_101a.yml │ └── view_101a.yml ├── stage1 │ ├── 101a_384.yml │ ├── 101a_384_recrop.yml │ ├── 101a_384_spgan.yml │ ├── densenet169a_384.yml │ ├── resnext101a_384.yml │ ├── s101_384.yml │ ├── se_resnet101a_384.yml │ └── transreid_256.yml └── stage2 │ ├── 101a_384.yml │ ├── 101a_384_recrop.yml │ ├── 101a_384_recrop_veri.yml │ ├── 101a_384_recrop_veri_gan.yml │ ├── 101a_384_spgan.yml │ ├── 101a_384_spgan_veri.yml │ ├── 101a_384_spgan_veri_gan.yml │ ├── 101a_384_veri.yml │ ├── 101a_384_veri_gan.yml │ ├── densenet169a_384.yml │ ├── densenet169a_384_veri.yml │ ├── densenet169a_384_veri_gan.yml │ ├── resnext101a_384.yml │ ├── resnext101a_384_veri.yml │ ├── resnext101a_384_veri_gan.yml │ ├── s101_384.yml │ ├── s101_384_veri.yml │ ├── s101_384_veri_gan.yml │ ├── se_resnet101a_384.yml │ ├── se_resnet101a_384_veri.yml │ ├── se_resnet101a_384_veri_gan.yml │ ├── transreid_256.yml │ ├── transreid_256_veri.yml │ └── transreid_256_veri_gan.yml ├── datasets ├── __init__.py ├── aic.py ├── aic_sim.py ├── aic_sim_spgan.py ├── autoaugment.py ├── bases.py ├── make_dataloader.py ├── preprocessing.py ├── sampler.py ├── sampler_ddp.py ├── sim_view.py ├── split_val.py └── veri.py ├── ensemble.py ├── loss ├── __init__.py ├── arcface.py ├── center_loss.py ├── make_loss.py ├── metric_learning.py ├── softmax_loss.py └── triplet_loss.py ├── model ├── __init__.py ├── backbones │ ├── __init__.py │ ├── densenet_ibn.py │ ├── resnest.py │ ├── resnet.py │ ├── resnet_ibn_a.py │ ├── resnext_ibn.py │ ├── se_module.py │ ├── se_resnet_ibn_a.py │ ├── transformer_block.py │ └── vit_pytorch.py ├── layers │ └── pooling.py └── make_model.py ├── processor ├── __init__.py ├── processor.py ├── processor_orig.py └── uda_processor.py ├── requirements.txt ├── run.sh ├── solver ├── __init__.py ├── lr_scheduler.py └── make_optimizer.py ├── test.py ├── test_orig.py ├── train.py ├── train_cam.py ├── train_stage2_v1.py ├── train_stage2_v2.py ├── train_view.py └── utils ├── __init__.py ├── faiss_rerank.py ├── faiss_utils.py ├── ficfac_torch.py ├── iotools.py ├── logger.py ├── meter.py ├── metrics.py ├── metrics_orig.py └── reranking.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/LICENSE -------------------------------------------------------------------------------- /PTGAN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/PTGAN.py -------------------------------------------------------------------------------- /PTGAN_for_one_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/PTGAN_for_one_test.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/README.md -------------------------------------------------------------------------------- /code_GAN/PTGAN_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/PTGAN_dataset.py -------------------------------------------------------------------------------- /code_GAN/dataset_for_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/dataset_for_test.py -------------------------------------------------------------------------------- /code_GAN/dist.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/dist.npy -------------------------------------------------------------------------------- /code_GAN/gan/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code_GAN/gan/contrastive_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/gan/contrastive_loss.py -------------------------------------------------------------------------------- /code_GAN/gan/gan_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/gan/gan_losses.py -------------------------------------------------------------------------------- /code_GAN/gan/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/gan/model.py -------------------------------------------------------------------------------- /code_GAN/gan/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/gan/networks.py -------------------------------------------------------------------------------- /code_GAN/gan/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/gan/options.py -------------------------------------------------------------------------------- /code_GAN/gan/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code_GAN/gan/utils/html.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/gan/utils/html.py -------------------------------------------------------------------------------- /code_GAN/gan/utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/gan/utils/util.py -------------------------------------------------------------------------------- /code_GAN/gan/utils/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/gan/utils/visualizer.py -------------------------------------------------------------------------------- /code_GAN/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/metrics.py -------------------------------------------------------------------------------- /code_GAN/orthogonal_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/orthogonal_encoder.py -------------------------------------------------------------------------------- /code_GAN/processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/processor.py -------------------------------------------------------------------------------- /code_GAN/processor_for_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/processor_for_test.py -------------------------------------------------------------------------------- /code_GAN/reid/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/__init__.py -------------------------------------------------------------------------------- /code_GAN/reid/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/datasets/__init__.py -------------------------------------------------------------------------------- /code_GAN/reid/datasets/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/datasets/train.py -------------------------------------------------------------------------------- /code_GAN/reid/evaluation_metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/evaluation_metrics/__init__.py -------------------------------------------------------------------------------- /code_GAN/reid/evaluation_metrics/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/evaluation_metrics/classification.py -------------------------------------------------------------------------------- /code_GAN/reid/evaluation_metrics/ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/evaluation_metrics/ranking.py -------------------------------------------------------------------------------- /code_GAN/reid/evaluators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/evaluators.py -------------------------------------------------------------------------------- /code_GAN/reid/feature_extraction/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/feature_extraction/__init__.py -------------------------------------------------------------------------------- /code_GAN/reid/feature_extraction/cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/feature_extraction/cnn.py -------------------------------------------------------------------------------- /code_GAN/reid/feature_extraction/database.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/feature_extraction/database.py -------------------------------------------------------------------------------- /code_GAN/reid/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/models/__init__.py -------------------------------------------------------------------------------- /code_GAN/reid/models/baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/models/baseline.py -------------------------------------------------------------------------------- /code_GAN/reid/models/embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/models/embedding.py -------------------------------------------------------------------------------- /code_GAN/reid/models/multi_branch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/models/multi_branch.py -------------------------------------------------------------------------------- /code_GAN/reid/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/models/resnet.py -------------------------------------------------------------------------------- /code_GAN/reid/trainers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/trainers.py -------------------------------------------------------------------------------- /code_GAN/reid/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/utils/__init__.py -------------------------------------------------------------------------------- /code_GAN/reid/utils/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/utils/data/__init__.py -------------------------------------------------------------------------------- /code_GAN/reid/utils/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/utils/data/dataset.py -------------------------------------------------------------------------------- /code_GAN/reid/utils/data/preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/utils/data/preprocessor.py -------------------------------------------------------------------------------- /code_GAN/reid/utils/data/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/utils/data/sampler.py -------------------------------------------------------------------------------- /code_GAN/reid/utils/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/utils/data/transforms.py -------------------------------------------------------------------------------- /code_GAN/reid/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/utils/logging.py -------------------------------------------------------------------------------- /code_GAN/reid/utils/meters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/utils/meters.py -------------------------------------------------------------------------------- /code_GAN/reid/utils/osutils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/utils/osutils.py -------------------------------------------------------------------------------- /code_GAN/reid/utils/serialization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/reid/utils/serialization.py -------------------------------------------------------------------------------- /code_GAN/test_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/test_processor.py -------------------------------------------------------------------------------- /code_GAN/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/train.py -------------------------------------------------------------------------------- /code_GAN/weights/GAN_stage_2/opt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/weights/GAN_stage_2/opt.txt -------------------------------------------------------------------------------- /code_GAN/weights/GAN_stage_2/web/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/code_GAN/weights/GAN_stage_2/web/index.html -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/config/__init__.py -------------------------------------------------------------------------------- /config/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/config/defaults.py -------------------------------------------------------------------------------- /configs/camera_view/camera_101a.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/camera_view/camera_101a.yml -------------------------------------------------------------------------------- /configs/camera_view/view_101a.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/camera_view/view_101a.yml -------------------------------------------------------------------------------- /configs/stage1/101a_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage1/101a_384.yml -------------------------------------------------------------------------------- /configs/stage1/101a_384_recrop.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage1/101a_384_recrop.yml -------------------------------------------------------------------------------- /configs/stage1/101a_384_spgan.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage1/101a_384_spgan.yml -------------------------------------------------------------------------------- /configs/stage1/densenet169a_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage1/densenet169a_384.yml -------------------------------------------------------------------------------- /configs/stage1/resnext101a_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage1/resnext101a_384.yml -------------------------------------------------------------------------------- /configs/stage1/s101_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage1/s101_384.yml -------------------------------------------------------------------------------- /configs/stage1/se_resnet101a_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage1/se_resnet101a_384.yml -------------------------------------------------------------------------------- /configs/stage1/transreid_256.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage1/transreid_256.yml -------------------------------------------------------------------------------- /configs/stage2/101a_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/101a_384.yml -------------------------------------------------------------------------------- /configs/stage2/101a_384_recrop.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/101a_384_recrop.yml -------------------------------------------------------------------------------- /configs/stage2/101a_384_recrop_veri.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/101a_384_recrop_veri.yml -------------------------------------------------------------------------------- /configs/stage2/101a_384_recrop_veri_gan.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/101a_384_recrop_veri_gan.yml -------------------------------------------------------------------------------- /configs/stage2/101a_384_spgan.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/101a_384_spgan.yml -------------------------------------------------------------------------------- /configs/stage2/101a_384_spgan_veri.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/101a_384_spgan_veri.yml -------------------------------------------------------------------------------- /configs/stage2/101a_384_spgan_veri_gan.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/101a_384_spgan_veri_gan.yml -------------------------------------------------------------------------------- /configs/stage2/101a_384_veri.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/101a_384_veri.yml -------------------------------------------------------------------------------- /configs/stage2/101a_384_veri_gan.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/101a_384_veri_gan.yml -------------------------------------------------------------------------------- /configs/stage2/densenet169a_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/densenet169a_384.yml -------------------------------------------------------------------------------- /configs/stage2/densenet169a_384_veri.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/densenet169a_384_veri.yml -------------------------------------------------------------------------------- /configs/stage2/densenet169a_384_veri_gan.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/densenet169a_384_veri_gan.yml -------------------------------------------------------------------------------- /configs/stage2/resnext101a_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/resnext101a_384.yml -------------------------------------------------------------------------------- /configs/stage2/resnext101a_384_veri.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/resnext101a_384_veri.yml -------------------------------------------------------------------------------- /configs/stage2/resnext101a_384_veri_gan.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/resnext101a_384_veri_gan.yml -------------------------------------------------------------------------------- /configs/stage2/s101_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/s101_384.yml -------------------------------------------------------------------------------- /configs/stage2/s101_384_veri.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/s101_384_veri.yml -------------------------------------------------------------------------------- /configs/stage2/s101_384_veri_gan.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/s101_384_veri_gan.yml -------------------------------------------------------------------------------- /configs/stage2/se_resnet101a_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/se_resnet101a_384.yml -------------------------------------------------------------------------------- /configs/stage2/se_resnet101a_384_veri.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/se_resnet101a_384_veri.yml -------------------------------------------------------------------------------- /configs/stage2/se_resnet101a_384_veri_gan.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/se_resnet101a_384_veri_gan.yml -------------------------------------------------------------------------------- /configs/stage2/transreid_256.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/transreid_256.yml -------------------------------------------------------------------------------- /configs/stage2/transreid_256_veri.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/transreid_256_veri.yml -------------------------------------------------------------------------------- /configs/stage2/transreid_256_veri_gan.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/configs/stage2/transreid_256_veri_gan.yml -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/aic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/datasets/aic.py -------------------------------------------------------------------------------- /datasets/aic_sim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/datasets/aic_sim.py -------------------------------------------------------------------------------- /datasets/aic_sim_spgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/datasets/aic_sim_spgan.py -------------------------------------------------------------------------------- /datasets/autoaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/datasets/autoaugment.py -------------------------------------------------------------------------------- /datasets/bases.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/datasets/bases.py -------------------------------------------------------------------------------- /datasets/make_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/datasets/make_dataloader.py -------------------------------------------------------------------------------- /datasets/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/datasets/preprocessing.py -------------------------------------------------------------------------------- /datasets/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/datasets/sampler.py -------------------------------------------------------------------------------- /datasets/sampler_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/datasets/sampler_ddp.py -------------------------------------------------------------------------------- /datasets/sim_view.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/datasets/sim_view.py -------------------------------------------------------------------------------- /datasets/split_val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/datasets/split_val.py -------------------------------------------------------------------------------- /datasets/veri.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/datasets/veri.py -------------------------------------------------------------------------------- /ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/ensemble.py -------------------------------------------------------------------------------- /loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/loss/__init__.py -------------------------------------------------------------------------------- /loss/arcface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/loss/arcface.py -------------------------------------------------------------------------------- /loss/center_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/loss/center_loss.py -------------------------------------------------------------------------------- /loss/make_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/loss/make_loss.py -------------------------------------------------------------------------------- /loss/metric_learning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/loss/metric_learning.py -------------------------------------------------------------------------------- /loss/softmax_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/loss/softmax_loss.py -------------------------------------------------------------------------------- /loss/triplet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/loss/triplet_loss.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/backbones/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/backbones/densenet_ibn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/model/backbones/densenet_ibn.py -------------------------------------------------------------------------------- /model/backbones/resnest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/model/backbones/resnest.py -------------------------------------------------------------------------------- /model/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/model/backbones/resnet.py -------------------------------------------------------------------------------- /model/backbones/resnet_ibn_a.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/model/backbones/resnet_ibn_a.py -------------------------------------------------------------------------------- /model/backbones/resnext_ibn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/model/backbones/resnext_ibn.py -------------------------------------------------------------------------------- /model/backbones/se_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/model/backbones/se_module.py -------------------------------------------------------------------------------- /model/backbones/se_resnet_ibn_a.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/model/backbones/se_resnet_ibn_a.py -------------------------------------------------------------------------------- /model/backbones/transformer_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/model/backbones/transformer_block.py -------------------------------------------------------------------------------- /model/backbones/vit_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/model/backbones/vit_pytorch.py -------------------------------------------------------------------------------- /model/layers/pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/model/layers/pooling.py -------------------------------------------------------------------------------- /model/make_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/model/make_model.py -------------------------------------------------------------------------------- /processor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/processor/__init__.py -------------------------------------------------------------------------------- /processor/processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/processor/processor.py -------------------------------------------------------------------------------- /processor/processor_orig.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/processor/processor_orig.py -------------------------------------------------------------------------------- /processor/uda_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/processor/uda_processor.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/run.sh -------------------------------------------------------------------------------- /solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/solver/__init__.py -------------------------------------------------------------------------------- /solver/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/solver/lr_scheduler.py -------------------------------------------------------------------------------- /solver/make_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/solver/make_optimizer.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/test.py -------------------------------------------------------------------------------- /test_orig.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/test_orig.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/train.py -------------------------------------------------------------------------------- /train_cam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/train_cam.py -------------------------------------------------------------------------------- /train_stage2_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/train_stage2_v1.py -------------------------------------------------------------------------------- /train_stage2_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/train_stage2_v2.py -------------------------------------------------------------------------------- /train_view.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/train_view.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/faiss_rerank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/utils/faiss_rerank.py -------------------------------------------------------------------------------- /utils/faiss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/utils/faiss_utils.py -------------------------------------------------------------------------------- /utils/ficfac_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/utils/ficfac_torch.py -------------------------------------------------------------------------------- /utils/iotools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/utils/iotools.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/utils/meter.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/metrics_orig.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/utils/metrics_orig.py -------------------------------------------------------------------------------- /utils/reranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ICCC-Platform/PTGAN/HEAD/utils/reranking.py --------------------------------------------------------------------------------