diff --git a/api/src/main/java/ai/djl/modality/cv/BufferedImageFactory.java b/api/src/main/java/ai/djl/modality/cv/BufferedImageFactory.java index 2a33510d5ab..16ece00dd0d 100644 --- a/api/src/main/java/ai/djl/modality/cv/BufferedImageFactory.java +++ b/api/src/main/java/ai/djl/modality/cv/BufferedImageFactory.java @@ -222,7 +222,7 @@ public NDArray toNDArray(NDManager manager, Flag flag) { int blue = rgb & 0xFF; if (flag == Flag.GRAYSCALE) { - int gray = (red + green + blue) / 3; + int gray = Math.round(0.299f * red + 0.587f * green + 0.114f * blue); bb.put((byte) gray); } else { bb.put((byte) red);