From 93143d7e3209e7da5bfd5afd673e282e35c56f88 Mon Sep 17 00:00:00 2001 From: WuHaobo Date: Tue, 28 Apr 2020 16:13:46 +0800 Subject: [PATCH] polish vgg --- configs/VGG/VGG13.yaml | 3 --- configs/VGG/VGG16.yaml | 3 --- configs/VGG/VGG19.yaml | 21 +++++++++++++++++++++ 3 files changed, 21 insertions(+), 6 deletions(-) diff --git a/configs/VGG/VGG13.yaml b/configs/VGG/VGG13.yaml index 67420fe13..dfd679bdc 100644 --- a/configs/VGG/VGG13.yaml +++ b/configs/VGG/VGG13.yaml @@ -48,8 +48,6 @@ TRAIN: order: '' - ToCHWImage: - - VALID: batch_size: 64 num_workers: 4 @@ -72,4 +70,3 @@ VALID: order: '' - ToCHWImage: - diff --git a/configs/VGG/VGG16.yaml b/configs/VGG/VGG16.yaml index 0e6648062..1862489bf 100644 --- a/configs/VGG/VGG16.yaml +++ b/configs/VGG/VGG16.yaml @@ -48,8 +48,6 @@ TRAIN: order: '' - ToCHWImage: - - VALID: batch_size: 64 num_workers: 4 @@ -72,4 +70,3 @@ VALID: order: '' - ToCHWImage: - diff --git a/configs/VGG/VGG19.yaml b/configs/VGG/VGG19.yaml index 3e4299791..e167539fe 100644 --- a/configs/VGG/VGG19.yaml +++ b/configs/VGG/VGG19.yaml @@ -48,4 +48,25 @@ TRAIN: order: '' - ToCHWImage: +VALID: + batch_size: 64 + num_workers: 4 + file_list: "./dataset/ILSVRC2012/val_list.txt" + data_dir: "./dataset/ILSVRC2012/" + shuffle_seed: 0 + transforms: + - DecodeImage: + to_rgb: True + to_np: False + channel_first: False + - ResizeImage: + resize_short: 256 + - CropImage: + size: 224 + - NormalizeImage: + scale: 1.0/255.0 + mean: [0.485, 0.456, 0.406] + std: [0.229, 0.224, 0.225] + order: '' + - ToCHWImage: