From eab1bc19e4a50a3f51bdc3e241b5f99db21df385 Mon Sep 17 00:00:00 2001 From: nblib Date: Wed, 9 Sep 2020 13:24:05 +0800 Subject: [PATCH] fix random erasing not work --- configs/DataAugment/ResNet50_RandomErasing.yaml | 4 ++-- .../quick_start/ResNet50_vd_ssld_random_erasing_finetune.yaml | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/configs/DataAugment/ResNet50_RandomErasing.yaml b/configs/DataAugment/ResNet50_RandomErasing.yaml index 682cf95d..995aa69f 100644 --- a/configs/DataAugment/ResNet50_RandomErasing.yaml +++ b/configs/DataAugment/ResNet50_RandomErasing.yaml @@ -49,13 +49,13 @@ TRAIN: mean: [0.485, 0.456, 0.406] std: [0.229, 0.224, 0.225] order: '' + - ToCHWImage: - RandomErasing: EPSILON: 0.5 sl: 0.02 sh: 0.4 - r1: 0.3 + r1: 0.3 mean: [0., 0., 0.] - - ToCHWImage: VALID: batch_size: 64 diff --git a/configs/quick_start/ResNet50_vd_ssld_random_erasing_finetune.yaml b/configs/quick_start/ResNet50_vd_ssld_random_erasing_finetune.yaml index 0687ea3b..019a05b4 100644 --- a/configs/quick_start/ResNet50_vd_ssld_random_erasing_finetune.yaml +++ b/configs/quick_start/ResNet50_vd_ssld_random_erasing_finetune.yaml @@ -48,9 +48,9 @@ TRAIN: mean: [0.485, 0.456, 0.406] std: [0.229, 0.224, 0.225] order: '' - - RandomErasing: - EPSILON: 0.5 - ToCHWImage: + - RandomErasing: + EPSILON: 0.5 VALID: batch_size: 20 -- GitLab