diff --git a/ppcls/configs/Pedestrian/strong_baseline_baseline.yaml b/ppcls/configs/Pedestrian/strong_baseline_baseline.yaml new file mode 100644 index 0000000000000000000000000000000000000000..886699f887bcc59983a58f5e41681a324ef90bc1 --- /dev/null +++ b/ppcls/configs/Pedestrian/strong_baseline_baseline.yaml @@ -0,0 +1,152 @@ +# global configs +Global: + checkpoints: null + pretrained_model: null + output_dir: "./output/" + device: "gpu" + save_interval: 1 + eval_during_train: True + eval_interval: 10 + epochs: 120 + print_batch_step: 10 + use_visualdl: False + # used for static mode and model export + image_shape: [3, 256, 128] + save_inference_dir: "./inference" + eval_mode: "retrieval" + +# model architecture +Arch: + name: "RecModel" + infer_output_key: "features" + infer_add_softmax: False + Backbone: + name: "ResNet50" + pretrained: True + stem_act: null + BackboneStopLayer: + name: "flatten" + Head: + name: "FC" + embedding_size: 2048 + class_num: 751 + +# loss function config for traing/eval process +Loss: + Train: + - CELoss: + weight: 1.0 + - TripletLossV2: + weight: 1.0 + margin: 0.3 + normalize_feature: false + Eval: + - CELoss: + weight: 1.0 + +Optimizer: + name: Adam + lr: + name: Piecewise + decay_epochs: [40, 70] + values: [0.00035, 0.000035, 0.0000035] + regularizer: + name: 'L2' + coeff: 0.0005 + +# data loader for train and eval +DataLoader: + Train: + dataset: + name: "VeriWild" + image_root: "./dataset/market1501" + cls_label_path: "./dataset/market1501/bounding_box_train.txt" + transform_ops: + - DecodeImage: + to_rgb: True + channel_first: False + - ResizeImage: + size: [128, 256] + - RandFlipImage: + flip_code: 1 + - Pad: + padding: 10 + - RandCropImage: + size: [128, 256] + - NormalizeImage: + scale: 0.00392157 + mean: [0.485, 0.456, 0.406] + std: [0.229, 0.224, 0.225] + order: '' + - RandomErasing: + EPSILON: 0.5 + sl: 0.02 + sh: 0.4 + r1: 0.3 + mean: [0.4914, 0.4822, 0.4465] + sampler: + name: PKSampler + batch_size: 64 + sample_per_id: 4 + drop_last: True + shuffle: True + loader: + num_workers: 6 + use_shared_memory: True + Eval: + Query: + dataset: + name: "VeriWild" + image_root: "./dataset/market1501" + cls_label_path: "./dataset/market1501/query.txt" + transform_ops: + - DecodeImage: + to_rgb: True + channel_first: False + - ResizeImage: + size: [128, 256] + - NormalizeImage: + scale: 0.00392157 + mean: [0.485, 0.456, 0.406] + std: [0.229, 0.224, 0.225] + order: '' + sampler: + name: DistributedBatchSampler + batch_size: 128 + drop_last: False + shuffle: False + loader: + num_workers: 6 + use_shared_memory: True + + Gallery: + dataset: + name: "VeriWild" + image_root: "./dataset/market1501" + cls_label_path: "./dataset/market1501/bounding_box_test.txt" + transform_ops: + - DecodeImage: + to_rgb: True + channel_first: False + - ResizeImage: + size: [128, 256] + - NormalizeImage: + scale: 0.00392157 + mean: [0.485, 0.456, 0.406] + std: [0.229, 0.224, 0.225] + order: '' + sampler: + name: DistributedBatchSampler + batch_size: 128 + drop_last: False + shuffle: False + loader: + num_workers: 6 + use_shared_memory: True + +Metric: + Eval: + - Recallk: + topk: [1, 5] + - mAP: {} + diff --git a/ppcls/configs/Pedestrian/reid_sb_ResNet50.yaml b/ppcls/configs/Pedestrian/strong_baseline_m1.yaml similarity index 98% rename from ppcls/configs/Pedestrian/reid_sb_ResNet50.yaml rename to ppcls/configs/Pedestrian/strong_baseline_m1.yaml index 8eb8da519f16b539e149bc946bd6c89e407ad35e..358fb18d458424451a71b3dafca79488d9bc1b18 100644 --- a/ppcls/configs/Pedestrian/reid_sb_ResNet50.yaml +++ b/ppcls/configs/Pedestrian/strong_baseline_m1.yaml @@ -6,7 +6,7 @@ Global: device: "gpu" save_interval: 1 eval_during_train: True - eval_interval: 1 + eval_interval: 10 epochs: 120 print_batch_step: 10 use_visualdl: False @@ -29,7 +29,7 @@ Arch: Head: name: "FC" embedding_size: 2048 - class_num: 1501 + class_num: 751 # loss function config for traing/eval process Loss: