mangye16 / ReID-Survey

Deep Learning for Person Re-identification: A Survey and Outlook
MIT License
626 stars 88 forks source link

训练部分遮挡数据集与论文中有偏差且较大 #13

Open hundunshu opened 3 years ago

hundunshu commented 3 years ago

您好,我在训练partial_reid数据集时,精度一直达不到论文的要求,相差还是蛮大的,

The test feature is normalized 2021-03-05 16:19:15,585 reid_baseline INFO: Validation Results - Epoch: 110 2021-03-05 16:19:15,585 reid_baseline INFO: mINP: 34.9% 2021-03-05 16:19:15,585 reid_baseline INFO: mAP: 50.2% 2021-03-05 16:19:15,585 reid_baseline INFO: CMC curve, Rank-1 :57.7% 2021-03-05 16:19:15,585 reid_baseline INFO: CMC curve, Rank-3 :66.0% 2021-03-05 16:19:15,585 reid_baseline INFO: CMC curve, Rank-5 :72.7% 2021-03-05 16:19:15,585 reid_baseline INFO: CMC curve, Rank-10 :80.3% The test feature is normalized 2021-03-05 16:19:16,431 reid_baseline INFO: Validation Results - Epoch: 110 2021-03-05 16:19:16,431 reid_baseline INFO: mINP: 54.8% 2021-03-05 16:19:16,431 reid_baseline INFO: mAP: 54.8% 2021-03-05 16:19:16,431 reid_baseline INFO: CMC curve, Rank-1 :42.9% 2021-03-05 16:19:16,431 reid_baseline INFO: CMC curve, Rank-3 :60.5% 2021-03-05 16:19:16,431 reid_baseline INFO: CMC curve, Rank-5 :65.5% 2021-03-05 16:19:16,431 reid_baseline INFO: CMC curve, Rank-10 :73.9% 配置: MODEL: PRETRAIN_CHOICE: 'imagenet' PRETRAIN_PATH: '../resnet50-19c8e357.pth' CENTER_LOSS: 'on' CENTER_FEAT_DIM: 2048 NAME: 'resnet50_nl' WEIGHT_REGULARIZED_TRIPLET: 'on' GENERALIZED_MEAN_POOL: 'on'

INPUT: IMG_SIZE: [256, 128] PROB: 0.5 # random horizontal flip RE_PROB: 0.5 # random erasing PADDING: 10

DATASETS: NAMES: ('market1501')

DATALOADER: PK_SAMPLER: 'on' NUM_INSTANCE: 4 NUM_WORKERS: 8

SOLVER: OPTIMIZER_NAME: 'Adam' MAX_EPOCHS: 120 BASE_LR: 0.00035

CENTER_LR: 0.5 CENTER_LOSS_WEIGHT: 0.0005

WEIGHT_DECAY: 0.0005 IMS_PER_BATCH: 64

STEPS: [40, 70] GAMMA: 0.1

WARMUP_FACTOR: 0.01 WARMUP_ITERS: 10 WARMUP_METHOD: 'linear'

CHECKPOINT_PERIOD: 40 LOG_PERIOD: 20 EVAL_PERIOD: 10

TEST: IMS_PER_BATCH: 128 RE_RANKING: 'off' WEIGHT: "path" FEAT_NORM: 'on' EVALUATE_ONLY: 'off' PARTIAL_REID: 'on'

OUTPUT_DIR: "../log/market1501/Experiment-AGW-baseline"

_C.DATASETS.ROOT_DIR = ('../dataset')

dataset文件夹下面有market1501数据集、partial_reid数据集、partial_ilids数据集

数据集是在给出的链接上下载得到的

请教大佬,这个是由于什么原因造成的呀?非常感谢!

Hellomodo commented 3 years ago

应该是pytorch和torchvision的版本没对上,确保这两个库的版本如下: pytorch=1.0.0 torchvision=0.2.1