keep trying to display the output

bioimage-io · Jan 2, 2025 · b6d3826 · b6d3826
1 parent 5765157
commit b6d3826
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/src/main/java/io/bioimage/modelrunner/tensorflow/v1/shm/ShmBuilder.java b/src/main/java/io/bioimage/modelrunner/tensorflow/v1/shm/ShmBuilder.java
@@ -124,14 +124,10 @@ private static void buildFromTensorFloat(Tensor<Float> tensor, String memoryName
         SharedMemoryArray shma = SharedMemoryArray.readOrCreate(memoryName, arrayShape, new FloatType(), false, true);
         ByteBuffer buff = shma.getDataBufferNoHeader();
         tensor.writeTo(buff);
+
         try (FileOutputStream fos = new FileOutputStream("/home/carlos/git/interp_out" + UUID.randomUUID().toString() + ".npy");
 	             FileChannel fileChannel = fos.getChannel()) {
-            	buff.rewind();
-	            // Write the buffer's content to the file
-	            while (buff.hasRemaining()) {
-	                fileChannel.write(buff);
-	            }
-	            buff.rewind();
+               fileChannel.write(ByteBuffer.wrap(tensor.bytesValue()));
 		} catch (IOException e) {
 			// TODO Auto-generated catch block
 			e.printStackTrace();