Bugfix: Update RGB model configurations' std from miscopied mean to i…

…magenet std
epic-kitchens · Dec 15, 2020 · d58e695 · d58e695
1 parent 5a50dfc
commit d58e695
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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: