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())