├── .gitignore ├── README.md ├── example_rank_lists_on_Duke ├── 00000126_0002_00000021.jpg ├── 00000147_0003_00000004.jpg ├── 00000169_0001_00000008.jpg ├── 00000257_0003_00000004.jpg ├── 00000288_0001_00000007.jpg ├── 00000300_0002_00000005.jpg ├── 00000570_0004_00000008.jpg ├── 00000611_0002_00000005.jpg ├── 00000641_0001_00000005.jpg ├── 00000849_0002_00000010.jpg ├── 00001228_0001_00000004.jpg ├── 00001463_0006_00000006.jpg ├── 00001782_0008_00000004.jpg ├── 00001901_0002_00000003.jpg ├── 00004331_0006_00000007.jpg └── 00004605_0006_00000007.jpg ├── requirements.txt ├── script ├── dataset │ ├── combine_trainval_sets.py │ ├── mapping_im_names_duke.py │ ├── mapping_im_names_market1501.py │ ├── transform_cuhk03.py │ ├── transform_duke.py │ └── transform_market1501.py └── experiment │ ├── infer_images_example.py │ ├── train.py │ └── visualize_rank_list.py └── tri_loss ├── __init__.py ├── dataset ├── Dataset.py ├── PreProcessImage.py ├── Prefetcher.py ├── TestSet.py ├── TrainSet.py └── __init__.py ├── model ├── Model.py ├── TripletLoss.py ├── __init__.py ├── loss.py └── resnet.py └── utils ├── __init__.py ├── dataset_utils.py ├── distance.py ├── metric.py ├── re_ranking.py ├── utils.py └── visualization.py /.gitignore: -------------------------------------------------------------------------------- 1 | .idea/ 2 | exp/ 3 | tmp/ 4 | *.pyc -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/README.md -------------------------------------------------------------------------------- /example_rank_lists_on_Duke/00000126_0002_00000021.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/example_rank_lists_on_Duke/00000126_0002_00000021.jpg -------------------------------------------------------------------------------- /example_rank_lists_on_Duke/00000147_0003_00000004.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/example_rank_lists_on_Duke/00000147_0003_00000004.jpg -------------------------------------------------------------------------------- /example_rank_lists_on_Duke/00000169_0001_00000008.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/example_rank_lists_on_Duke/00000169_0001_00000008.jpg -------------------------------------------------------------------------------- /example_rank_lists_on_Duke/00000257_0003_00000004.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/example_rank_lists_on_Duke/00000257_0003_00000004.jpg -------------------------------------------------------------------------------- /example_rank_lists_on_Duke/00000288_0001_00000007.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/example_rank_lists_on_Duke/00000288_0001_00000007.jpg -------------------------------------------------------------------------------- /example_rank_lists_on_Duke/00000300_0002_00000005.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/example_rank_lists_on_Duke/00000300_0002_00000005.jpg -------------------------------------------------------------------------------- /example_rank_lists_on_Duke/00000570_0004_00000008.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/example_rank_lists_on_Duke/00000570_0004_00000008.jpg -------------------------------------------------------------------------------- /example_rank_lists_on_Duke/00000611_0002_00000005.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/example_rank_lists_on_Duke/00000611_0002_00000005.jpg -------------------------------------------------------------------------------- /example_rank_lists_on_Duke/00000641_0001_00000005.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/example_rank_lists_on_Duke/00000641_0001_00000005.jpg -------------------------------------------------------------------------------- /example_rank_lists_on_Duke/00000849_0002_00000010.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/example_rank_lists_on_Duke/00000849_0002_00000010.jpg -------------------------------------------------------------------------------- /example_rank_lists_on_Duke/00001228_0001_00000004.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/example_rank_lists_on_Duke/00001228_0001_00000004.jpg -------------------------------------------------------------------------------- /example_rank_lists_on_Duke/00001463_0006_00000006.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/example_rank_lists_on_Duke/00001463_0006_00000006.jpg -------------------------------------------------------------------------------- /example_rank_lists_on_Duke/00001782_0008_00000004.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/example_rank_lists_on_Duke/00001782_0008_00000004.jpg -------------------------------------------------------------------------------- /example_rank_lists_on_Duke/00001901_0002_00000003.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/example_rank_lists_on_Duke/00001901_0002_00000003.jpg -------------------------------------------------------------------------------- /example_rank_lists_on_Duke/00004331_0006_00000007.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/example_rank_lists_on_Duke/00004331_0006_00000007.jpg -------------------------------------------------------------------------------- /example_rank_lists_on_Duke/00004605_0006_00000007.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/example_rank_lists_on_Duke/00004605_0006_00000007.jpg -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/requirements.txt -------------------------------------------------------------------------------- /script/dataset/combine_trainval_sets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/script/dataset/combine_trainval_sets.py -------------------------------------------------------------------------------- /script/dataset/mapping_im_names_duke.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/script/dataset/mapping_im_names_duke.py -------------------------------------------------------------------------------- /script/dataset/mapping_im_names_market1501.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/script/dataset/mapping_im_names_market1501.py -------------------------------------------------------------------------------- /script/dataset/transform_cuhk03.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/script/dataset/transform_cuhk03.py -------------------------------------------------------------------------------- /script/dataset/transform_duke.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/script/dataset/transform_duke.py -------------------------------------------------------------------------------- /script/dataset/transform_market1501.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/script/dataset/transform_market1501.py -------------------------------------------------------------------------------- /script/experiment/infer_images_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/script/experiment/infer_images_example.py -------------------------------------------------------------------------------- /script/experiment/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/script/experiment/train.py -------------------------------------------------------------------------------- /script/experiment/visualize_rank_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/script/experiment/visualize_rank_list.py -------------------------------------------------------------------------------- /tri_loss/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tri_loss/dataset/Dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/tri_loss/dataset/Dataset.py -------------------------------------------------------------------------------- /tri_loss/dataset/PreProcessImage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/tri_loss/dataset/PreProcessImage.py -------------------------------------------------------------------------------- /tri_loss/dataset/Prefetcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/tri_loss/dataset/Prefetcher.py -------------------------------------------------------------------------------- /tri_loss/dataset/TestSet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/tri_loss/dataset/TestSet.py -------------------------------------------------------------------------------- /tri_loss/dataset/TrainSet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/tri_loss/dataset/TrainSet.py -------------------------------------------------------------------------------- /tri_loss/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/tri_loss/dataset/__init__.py -------------------------------------------------------------------------------- /tri_loss/model/Model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/tri_loss/model/Model.py -------------------------------------------------------------------------------- /tri_loss/model/TripletLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/tri_loss/model/TripletLoss.py -------------------------------------------------------------------------------- /tri_loss/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tri_loss/model/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/tri_loss/model/loss.py -------------------------------------------------------------------------------- /tri_loss/model/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/tri_loss/model/resnet.py -------------------------------------------------------------------------------- /tri_loss/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tri_loss/utils/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/tri_loss/utils/dataset_utils.py -------------------------------------------------------------------------------- /tri_loss/utils/distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/tri_loss/utils/distance.py -------------------------------------------------------------------------------- /tri_loss/utils/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/tri_loss/utils/metric.py -------------------------------------------------------------------------------- /tri_loss/utils/re_ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/tri_loss/utils/re_ranking.py -------------------------------------------------------------------------------- /tri_loss/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/tri_loss/utils/utils.py -------------------------------------------------------------------------------- /tri_loss/utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/person-reid-triplet-loss-baseline/HEAD/tri_loss/utils/visualization.py --------------------------------------------------------------------------------