Resolve mean/std swap for VITDet backbone (keras-team#2087)

yuvraj-wale · Feb 8, 2024 · 3feef5b · 3feef5b
1 parent bec2706
commit 3feef5b
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 2 deletions.
diff --git a/keras_cv/models/backbones/vit_det/data/vitdet_base_out.npz b/keras_cv/models/backbones/vit_det/data/vitdet_base_out.npz
diff --git a/keras_cv/models/backbones/vit_det/vit_det_backbone.py b/keras_cv/models/backbones/vit_det/vit_det_backbone.py
@@ -125,8 +125,8 @@ def __init__(
             x = keras.layers.Rescaling(1.0 / 255.0)(x)
 
         # VITDet scales inputs based on the standard ImageNet mean/stddev.
-        x = (x - ops.array([0.229, 0.224, 0.225], dtype=x.dtype)) / (
-            ops.array([0.485, 0.456, 0.406], dtype=x.dtype)
+        x = (x - ops.array([0.485, 0.456, 0.406], dtype=x.dtype)) / (
+            ops.array([0.229, 0.224, 0.225], dtype=x.dtype)
         )
 
         x = ViTDetPatchingAndEmbedding(

diff --git a/keras_cv/models/segmentation/segment_anything/data/sam_base_out_iou_pred.npy b/keras_cv/models/segmentation/segment_anything/data/sam_base_out_iou_pred.npy
diff --git a/keras_cv/models/segmentation/segment_anything/data/sam_base_out_masks.npy b/keras_cv/models/segmentation/segment_anything/data/sam_base_out_masks.npy