diff --git 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 index f46d7be190bae519cefd1f15a0c5d45b44993c91..5501dd903bce06d37c3e613057f793b5916ef613 100644 --- 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: