estoy muy seco

bioimage-io · Nov 23, 2024 · b006f0b · b006f0b
1 parent efe7c4f
commit b006f0b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/shm/ShmBuilder.java b/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/shm/ShmBuilder.java
@@ -111,11 +111,14 @@ private static void buildFromTensorFloat(Tensor tensor, String memoryName) throw
 			throw new IllegalArgumentException("Model output tensor with shape " + Arrays.toString(arrayShape) 
 					+ " is too big. Max number of elements per float output tensor supported: " + Integer.MAX_VALUE / 4);
 		SharedMemoryArray shma = SharedMemoryArray.readOrCreate(memoryName, arrayShape, new FloatType(), false, true);
-		long flatSize = 1;
+		/*long flatSize = 1;
     	for (long l : arrayShape) {flatSize *= l;}
     	ByteBuffer byteBuffer = ByteBuffer.allocate((int) (flatSize * Float.BYTES));
     	tensor.data_ptr_float().get(byteBuffer.asFloatBuffer().array());
         shma.getDataBufferNoHeader().put(byteBuffer);
+        */
+		RandomAccessibleInterval<?> rai = shma.getSharedRAI();
+		rai = ImgLib2Builder.build(tensor);
         if (PlatformDetection.isWindows()) shma.close();
     }