Match OpenCV to grayscale algorithm, deepjavalibrary#964

Change-Id: I3e7ae2731d4ad36e3cbe27ec4a2ebd0bc3749ccd
frankfliu · May 18, 2021 · 48d9fbf · 48d9fbf
1 parent c17c15a
commit 48d9fbf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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);