No public description

PiperOrigin-RevId: 606319053
tensorflow · Feb 12, 2024 · f5146ac · f5146ac
1 parent c0abb86
commit f5146ac
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/official/vision/serving/semantic_segmentation.py b/official/vision/serving/semantic_segmentation.py
@@ -37,8 +37,11 @@ def _build_inputs(self, image):
     """Builds classification model inputs for serving."""
 
     # Normalizes image with mean and std pixel values.
+    image_feature = self.params.task.train_data.image_feature
     image = preprocess_ops.normalize_image(
-        image, offset=preprocess_ops.MEAN_RGB, scale=preprocess_ops.STDDEV_RGB)
+        image,
+        offset=image_feature.mean,
+        scale=image_feature.stddev)
 
     if self.params.task.train_data.preserve_aspect_ratio:
       image, image_info = preprocess_ops.resize_and_crop_image(