From d58e69562affdcdc1e40607ebaa0a954d4034cbf Mon Sep 17 00:00:00 2001 From: Will Price Date: Tue, 15 Dec 2020 22:27:33 +0000 Subject: [PATCH] Bugfix: Update RGB model configurations' std from miscopied mean to imagenet std --- configs/trn_rgb.yaml | 2 +- configs/tsm_rgb.yaml | 2 +- configs/tsn_rgb.yaml | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/configs/trn_rgb.yaml b/configs/trn_rgb.yaml index 952d359..4707a7f 100644 --- a/configs/trn_rgb.yaml +++ b/configs/trn_rgb.yaml @@ -18,7 +18,7 @@ data: input_size: 224 scale_size: 256 mean: [0.485, 0.456, 0.406] - std: [0.485, 0.456, 0.406] + std: [0.229, 0.224, 0.225] train_augmentation: multiscale_crop_scales: [1, 0.875, 0.75, 0.66] test_augmentation: diff --git a/configs/tsm_rgb.yaml b/configs/tsm_rgb.yaml index e35f6b7..ffe3484 100644 --- a/configs/tsm_rgb.yaml +++ b/configs/tsm_rgb.yaml @@ -18,7 +18,7 @@ data: input_size: 224 scale_size: 256 mean: [0.485, 0.456, 0.406] - std: [0.485, 0.456, 0.406] + std: [0.229, 0.224, 0.225] train_augmentation: multiscale_crop_scales: [1, 0.875, 0.75, 0.66] test_augmentation: diff --git a/configs/tsn_rgb.yaml b/configs/tsn_rgb.yaml index 0af8001..a90cd4d 100644 --- a/configs/tsn_rgb.yaml +++ b/configs/tsn_rgb.yaml @@ -19,7 +19,7 @@ data: input_size: 224 scale_size: 256 mean: [0.485, 0.456, 0.406] - std: [0.485, 0.456, 0.406] + std: [0.229, 0.224, 0.225] train_augmentation: multiscale_crop_scales: [1, 0.875, 0.75, 0.66] test_augmentation: