diff --git a/ppcls/configs/Logo/ResNet50_ReID.yaml b/ppcls/configs/Logo/ResNet50_ReID.yaml index 0949add860ccbbabe594953c4e1f6fd6fdf1bb5e..bfbedf8f982dc3e808e0ef0e031254c28911d95a 100644 --- a/ppcls/configs/Logo/ResNet50_ReID.yaml +++ b/ppcls/configs/Logo/ResNet50_ReID.yaml @@ -32,7 +32,7 @@ Arch: Head: name: "CircleMargin" margin: 0.35 - scale: 64 + scale: 64 embedding_size: 512 class_num: 3000 @@ -56,48 +56,74 @@ Optimizer: name: Cosine learning_rate: 0.04 regularizer: - name: 'L2' + name: "L2" coeff: 0.0001 # data loader for train and eval DataLoader: Train: dataset: + name: ImageNetDataset + image_root: "dataset/LogoDet-3K-crop/train/" + cls_label_path: "dataset/LogoDet-3K-crop/train_list.txt" + relabel: True + transform_ops: + - DecodeImage: + to_rgb: True + channel_first: False + - ResizeImage: + size: 224 + - RandFlipImage: + flip_code: 1 + - AugMix: + prob: 0.5 + - NormalizeImage: + scale: 0.00392157 + mean: [0.485, 0.456, 0.406] + std: [0.229, 0.224, 0.225] + order: "" + - RandomErasing: + EPSILON: 0.5 + sampler: + name: PKSampler + batch_size: 128 + sample_per_id: 2 + drop_last: True + + loader: + num_workers: 6 + use_shared_memory: True + Eval: + Query: + dataset: name: ImageNetDataset - image_root: "dataset/LogoDet-3K-crop/train/" - cls_label_path: "dataset/LogoDet-3K-crop/train_list.txt" + image_root: "dataset/LogoDet-3K-crop/val/" + cls_label_path: "dataset/LogoDet-3K-crop/query_list.txt" transform_ops: - DecodeImage: to_rgb: True channel_first: False - ResizeImage: size: 224 - - RandFlipImage: - flip_code: 1 - - AugMix: - prob: 0.5 - NormalizeImage: scale: 0.00392157 mean: [0.485, 0.456, 0.406] std: [0.229, 0.224, 0.225] - order: '' - - RandomErasing: - EPSILON: 0.5 - sampler: - name: PKSampler + order: "" + sampler: + name: DistributedBatchSampler batch_size: 128 - sample_per_id: 2 - drop_last: True - - loader: - num_workers: 6 + drop_last: False + shuffle: False + loader: + num_workers: 8 use_shared_memory: True - Eval: - Query: + + Gallery: dataset: name: ImageNetDataset - image_root: "dataset/LogoDet-3K-crop/val/" - cls_label_path: "dataset/LogoDet-3K-crop/query_list.txt" + image_root: "dataset/LogoDet-3K-crop/train/" + cls_label_path: "dataset/LogoDet-3K-crop/train_list.txt" transform_ops: - DecodeImage: to_rgb: True @@ -108,44 +134,18 @@ DataLoader: scale: 0.00392157 mean: [0.485, 0.456, 0.406] std: [0.229, 0.224, 0.225] - order: '' + order: "" sampler: - name: DistributedBatchSampler - batch_size: 128 - drop_last: False - shuffle: False - loader: - num_workers: 8 - use_shared_memory: True - - Gallery: - dataset: - name: ImageNetDataset - image_root: "dataset/LogoDet-3K-crop/train/" - cls_label_path: "dataset/LogoDet-3K-crop/train_list.txt" - transform_ops: - - DecodeImage: - to_rgb: True - channel_first: False - - ResizeImage: - size: 224 - - 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 + name: DistributedBatchSampler + batch_size: 128 + drop_last: False + shuffle: False loader: - num_workers: 8 - use_shared_memory: True + num_workers: 8 + use_shared_memory: True Metric: Eval: - Recallk: topk: [1, 5] - mAP: {} -