Update mv3_large_x1_0_distill_mv3_small_x1_0.yaml

25cd0f9e · cuicheng01 · GitHub · 3e30e4a8 · 25cd0f9e
显示空白变更内容
内联并排

Showing with 6 addition and 0 deletion

ppcls/configs/ImageNet/Distillation/mv3_large_x1_0_distill_mv3_small_x1_0.yaml ...t/Distillation/mv3_large_x1_0_distill_mv3_small_x1_0.yaml +6 -0

未找到文件。
--- a/ppcls/configs/ImageNet/Distillation/mv3_large_x1_0_distill_mv3_small_x1_0.yaml
+++ b/ppcls/configs/ImageNet/Distillation/mv3_large_x1_0_distill_mv3_small_x1_0.yaml
@@ -67,6 +67,9 @@ DataLoader:
        image_root: "./dataset/ILSVRC2012/"
        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
        transform_ops:
+          - DecodeImage:
+              to_rgb: True
+              channel_first: False
          - RandCropImage:
              size: 224
          - RandFlipImage:
@@ -94,6 +97,9 @@ DataLoader:
        image_root: "./dataset/ILSVRC2012/"
        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
        transform_ops:
+          - DecodeImage:
+              to_rgb: True
+              channel_first: False
          - ResizeImage:
              resize_short: 256
          - CropImage: