Update configs and README

README.md

@@ -54,6 +54,7 @@ Res2Net200_vd预训练模型Top-1精度高达85.1%。
     - [主体检测](./docs/zh_CN/application/mainbody_detection.md)
     - 图像分类
         - [ImageNet分类任务](./docs/zh_CN/tutorials/quick_start_professional.md)
+    - [特征学习](./docs/zh_CN/application/feature_learning.md)
         - [特征学习](./docs/zh_CN/application/feature_learning.md)
         - [商品识别](./docs/zh_CN/application/product_recognition.md)
         - [车辆识别](./docs/zh_CN/application/vehicle_recognition.md)
@@ -62,7 +63,7 @@ Res2Net200_vd预训练模型Top-1精度高达85.1%。
     - [向量检索](./deploy/vector_search/README.md)
 - 模型训练/评估
     - [图像分类任务](./docs/zh_CN/tutorials/getting_started.md)
-    - [特征学习任务](./docs/zh_CN/application/feature_learning.md)
+    - [特征学习任务](./docs/zh_CN/tutorials/getting_started_retrieval.md)
 - 模型预测（当前只支持图像分类任务，图像识别更新中）
     - [基于Python预测引擎预测推理](./docs/zh_CN/tutorials/getting_started.md)
     - [基于C++预测引擎预测推理](./deploy/cpp_infer/readme.md)

ppcls/configs/Cartoonface/ResNet50_icartoon.yaml

@@ -84,7 +84,7 @@ DataLoader:
         shuffle: True
     loader:
         num_workers: 6
-        use_shared_memory: False
+        use_shared_memory: True
   
   Eval:
     Query:

ppcls/configs/Logo/ResNet50_ReID.yaml

@@ -92,7 +92,7 @@ DataLoader:
 
     loader:
         num_workers: 6
-        use_shared_memory: False
+        use_shared_memory: True
   Eval:
     Query:
       dataset:

ppcls/configs/Products/ResNet50_vd_Aliproduct.yaml

@@ -66,7 +66,7 @@ DataLoader:
         - DecodeImage:
             to_rgb: True
             channel_first: False
-        - ResizeImage:
+        - RandCropImage:
             size: 224
         - RandFlipImage:
             flip_code: 1

ppcls/configs/Products/ResNet50_vd_SOP.yaml

@@ -69,7 +69,7 @@ Optimizer:
 DataLoader:
   Train:
     dataset:
-      name: ImageNetDataset
+      name: VeriWild
       image_root: ./dataset/Stanford_Online_Products/
       cls_label_path: ./dataset/Stanford_Online_Products/train_list.txt
       transform_ops:
@@ -104,7 +104,7 @@ DataLoader:
   Eval:
     Query:
       dataset: 
-        name: ImageNetDataset
+        name: VeriWild
         image_root: ./dataset/Stanford_Online_Products/
         cls_label_path: ./dataset/Stanford_Online_Products/test_list.txt
         transform_ops:
@@ -129,7 +129,7 @@ DataLoader:
 
     Gallery:
       dataset: 
-        name: ImageNetDataset
+        name: VeriWild
         image_root: ./dataset/Stanford_Online_Products/
         cls_label_path: ./dataset/Stanford_Online_Products/test_list.txt
         transform_ops:

ppcls/configs/Vehicle/ResNet50_ReID.yaml

@@ -100,7 +100,7 @@ DataLoader:
         shuffle: True
     loader:
         num_workers: 6
-        use_shared_memory: False
+        use_shared_memory: True
   Eval:
     Query:
       dataset: 
@@ -125,7 +125,7 @@ DataLoader:
         shuffle: False
       loader:
         num_workers: 6
-        use_shared_memory: False
+        use_shared_memory: True
 
     Gallery:
       dataset: 
@@ -150,7 +150,7 @@ DataLoader:
         shuffle: False
       loader:
         num_workers: 6
-        use_shared_memory: False
+        use_shared_memory: True
 
 Metric:
   Eval:

ppcls/configs/quick_start/ResNet50_vd_finetune_retrieval.yaml

-Global:
-  checkpoints: null
-  pretrained_model: null
-  output_dir: "./output/"
-  device: "gpu"
-  class_num: 102
-  save_interval: 1
-  eval_mode: "retrieval"
-  eval_during_train: True
-  eval_interval: 1
-  epochs: 20
-  print_batch_step: 10
-  use_visualdl: False
-  image_shape: [3, 224, 224]
-
-  #inference related
-  save_inference_dir: "./inference"
-
-Arch:
-  name: "RecModel"
-  infer_output_key:  "features"
-  infer_add_softmax: "false"
-  Backbone:
-    name: "ResNet50_vd"
-    pretrained: False
-  BackboneStopLayer: 
-    name: "flatten_0"
-    output_dim: 2048
-  Head:
-    name: "FC"
-    class_num: 102
-    embedding_size: 2048
-
-Loss:
-  Train:
-    - CELoss:
-        weight: 1.0
-  Eval:
-    - CELoss:
-        weight: 1.0
-
-Optimizer:
-  name: Momentum
-  momentum: 0.9
-  lr:
-    name: Piecewise
-    learning_rate: 0.1
-    decay_epochs: [30, 60, 90]
-    values: [0.1, 0.01, 0.001, 0.0001]
-  regularizer:
-    name: 'L2'
-    coeff: 0.0001
-
-DataLoader:
-  Train:
-    dataset:
-        name: ImageNetDataset
-        image_root:  "./dataset/flowers102/"
-        cls_label_path:  "./dataset/flowers102/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
-    sampler:
-        name: DistributedBatchSampler
-        batch_size: 256
-        drop_last: False
-        shuffle: True
-    loader:
-        num_workers: 6
-        use_shared_memory: False
-  
-  Eval:
-    Query:
-      dataset: 
-          name: ImageNetDataset
-          image_root: "./dataset/flowers102/"
-          cls_label_path: "./dataset/flowers102/val_list.txt"
-          transform_ops:
-            - ResizeImage:
-                resize_short: 256
-            - CropImage:
-                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: 512
-          drop_last: False
-          shuffle: False
-      loader:
-          num_workers: 6
-          use_shared_memory: True
-
-    Gallery:
-      dataset: 
-          name: ImageNetDataset
-          image_root: "./dataset/flowers102/"
-          cls_label_path: "./dataset/flowers102/train_list.txt"
-          transform_ops:
-            - ResizeImage:
-                resize_short: 256
-            - CropImage:
-                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: 512
-          drop_last: False
-          shuffle: False
-      loader:
-          num_workers: 6
-          use_shared_memory: True
-
-Metric:
-    Train:
-    - TopkAcc:
-        topk: [1, 5]
-    Eval:
-    - Recallk:
-        topk: [1, 10]
-