From b01f47a092f77c39d140cab74671f2e7f6fa21b9 Mon Sep 17 00:00:00 2001 From: panfengfeng Date: Mon, 29 Jun 2020 20:53:04 +0800 Subject: [PATCH] fix mobilenetv3 scripts --- model_zoo/mobilenetv3/src/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model_zoo/mobilenetv3/src/dataset.py b/model_zoo/mobilenetv3/src/dataset.py index ae5070e359..d2fa12f793 100644 --- a/model_zoo/mobilenetv3/src/dataset.py +++ b/model_zoo/mobilenetv3/src/dataset.py @@ -62,7 +62,7 @@ def create_dataset(dataset_path, do_train, config, platform, repeat_num=1, batch resize_crop_op = C.RandomCropDecodeResize(resize_height, scale=(0.08, 1.0), ratio=(0.75, 1.333)) horizontal_flip_op = C.RandomHorizontalFlip(prob=0.5) - resize_op = C.Resize((256, 256)) + resize_op = C.Resize(256) center_crop = C.CenterCrop(resize_width) rescale_op = C.RandomColorAdjust(brightness=0.4, contrast=0.4, saturation=0.4) normalize_op = C.Normalize(mean=[0.485*255, 0.456*255, 0.406*255], std=[0.229*255, 0.224*255, 0.225*255])