From 126246832b4dcc7fa5ae0332d85bc190f9178df6 Mon Sep 17 00:00:00 2001 From: cuicheng01 Date: Thu, 23 Sep 2021 02:51:50 +0000 Subject: [PATCH] Update GeneralRecognition_PPLCNet_x2_5.yaml --- .../GeneralRecognition_LCNet_x2_5.yaml | 149 ------------------ 1 file changed, 149 deletions(-) delete mode 100644 ppcls/configs/GeneralRecognition/GeneralRecognition_LCNet_x2_5.yaml diff --git a/ppcls/configs/GeneralRecognition/GeneralRecognition_LCNet_x2_5.yaml b/ppcls/configs/GeneralRecognition/GeneralRecognition_LCNet_x2_5.yaml deleted file mode 100644 index a7576b74..00000000 --- a/ppcls/configs/GeneralRecognition/GeneralRecognition_LCNet_x2_5.yaml +++ /dev/null @@ -1,149 +0,0 @@ -# global configs -Global: - checkpoints: null - pretrained_model: null - output_dir: ./output/ - device: gpu - save_interval: 1 - eval_during_train: True - eval_interval: 1 - epochs: 100 - print_batch_step: 10 - use_visualdl: False - # used for static mode and model export - image_shape: [3, 224, 224] - save_inference_dir: ./inference - eval_mode: retrieval - use_dali: False - to_static: False - -# model architecture -Arch: - name: RecModel - infer_output_key: features - infer_add_softmax: False - - Backbone: - name: LCNet_x2_5 - pretrained: True - use_ssld: True - BackboneStopLayer: - name: flatten_0 - Neck: - name: FC - embedding_size: 1280 - class_num: 512 - Head: - name: ArcMargin - embedding_size: 512 - class_num: 185341 - margin: 0.2 - scale: 30 - -# loss function config for traing/eval process -Loss: - Train: - - CELoss: - weight: 1.0 - Eval: - - CELoss: - weight: 1.0 - -Optimizer: - name: Momentum - momentum: 0.9 - lr: - name: Cosine - learning_rate: 0.04 - warmup_epoch: 5 - regularizer: - name: 'L2' - coeff: 0.00001 - - -# data loader for train and eval -DataLoader: - Train: - dataset: - name: ImageNetDataset - image_root: ./dataset/ - cls_label_path: ./dataset/train_reg_all_data.txt - transform_ops: - - DecodeImage: - to_rgb: True - channel_first: False - - RandCropImage: - size: 224 - - RandFlipImage: - flip_code: 1 - - NormalizeImage: - scale: 1.0/255.0 - mean: [0.485, 0.456, 0.406] - std: [0.229, 0.224, 0.225] - order: '' - - sampler: - name: DistributedBatchSampler - batch_size: 128 - num_instances: 2 - drop_last: False - shuffle: True - loader: - num_workers: 4 - use_shared_memory: True - - Eval: - Query: - dataset: - name: VeriWild - image_root: ./dataset/Aliproduct/ - cls_label_path: ./dataset/Aliproduct/val_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: 64 - drop_last: False - shuffle: False - loader: - num_workers: 4 - use_shared_memory: True - - Gallery: - dataset: - name: VeriWild - image_root: ./dataset/Aliproduct/ - cls_label_path: ./dataset/Aliproduct/val_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: 64 - drop_last: False - shuffle: False - loader: - num_workers: 4 - use_shared_memory: True - -Metric: - Eval: - - Recallk: - topk: [1, 5] -- GitLab