From 1f750b06e24f0631db52a004b1a158ea8bc63ee8 Mon Sep 17 00:00:00 2001 From: John Doll Date: Tue, 5 Jul 2022 12:52:43 -0700 Subject: [PATCH] variable naming updates --- .../djl/examples/inference/SemanticSegmentation.java | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/examples/src/main/java/ai/djl/examples/inference/SemanticSegmentation.java b/examples/src/main/java/ai/djl/examples/inference/SemanticSegmentation.java index b7ff0f11e8e..3f6633e6cc5 100644 --- a/examples/src/main/java/ai/djl/examples/inference/SemanticSegmentation.java +++ b/examples/src/main/java/ai/djl/examples/inference/SemanticSegmentation.java @@ -66,12 +66,12 @@ public static void predict() throws IOException, ModelException, TranslateExcept } // get image dimensions - final int height = img.getHeight(); - final int width = img.getWidth(); + int height = img.getHeight(); + int width = img.getWidth(); // rgb coloring means and standard deviations for improved detection performance - final float[] MEAN = {0.485f, 0.456f, 0.406f}; - final float[] STD = {0.229f, 0.224f, 0.225f}; + float[] mean = {0.485f, 0.456f, 0.406f}; + float[] std = {0.229f, 0.224f, 0.225f}; Criteria criteria = Criteria.builder() @@ -82,7 +82,7 @@ public static void predict() throws IOException, ModelException, TranslateExcept SemanticSegmentationTranslator.builder() .addTransform(new Resize(width, height)) .addTransform(new ToTensor()) - .addTransform(new Normalize(MEAN, STD)) + .addTransform(new Normalize(mean, std)) .optSynsetUrl( "https://mlrepo.djl.ai/model/cv/instance_segmentation/ai/djl/mxnet/mask_rcnn/classes.txt") .build())