Make BufferedImageFactory.fromNDArray synchronous (#1339)

Found that it gives some performance improvement.
deepjavalibrary · Nov 2, 2021 · 518cadc · 518cadc
1 parent 11d4013
commit 518cadc
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/api/src/main/java/ai/djl/modality/cv/BufferedImageFactory.java b/api/src/main/java/ai/djl/modality/cv/BufferedImageFactory.java
@@ -108,7 +108,6 @@ public Image fromNDArray(NDArray array) {
             BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
             int[] raw = array.toUint8Array();
             IntStream.range(0, imageArea)
-                    .parallel()
                     .forEach(
                             ele -> {
                                 int x = ele % width;
@@ -127,7 +126,6 @@ public Image fromNDArray(NDArray array) {
             BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
             int[] raw = array.toUint8Array();
             IntStream.range(0, imageArea)
-                    .parallel()
                     .forEach(
                             ele -> {
                                 int x = ele % width;