keep tracking the output

Author: carlosuc3m

src/main/java/io/bioimage/modelrunner/tensorflow/v1/Tensorflow1Interface.java

@@ -34,6 +34,7 @@
 import io.bioimage.modelrunner.engine.EngineInfo;
 import io.bioimage.modelrunner.exceptions.LoadModelException;
 import io.bioimage.modelrunner.exceptions.RunModelException;
+import io.bioimage.modelrunner.numpy.DecodeNumpy;
 import io.bioimage.modelrunner.system.PlatformDetection;
 import io.bioimage.modelrunner.tensor.Tensor;
 import io.bioimage.modelrunner.tensor.shm.SharedMemoryArray;
@@ -63,6 +64,7 @@
 import java.util.LinkedList;
 import java.util.List;
 import java.util.Map;
+import java.util.UUID;
 import java.util.stream.Collectors;
 
 import org.tensorflow.SavedModelBundle;
@@ -394,7 +396,8 @@ else if (task.status == TaskStatus.CRASHED) {
 	        		shm = SharedMemoryArray.read(name);
 	        		shmaOutputList.add(shm);
 	        	}
-	        	RandomAccessibleInterval<?> rai = shm.getSharedRAI();
+	        	RandomAccessibleInterval<T> rai = shm.getSharedRAI();
+	        	DecodeNumpy.saveNpy("/home/carlos/git/out_" + UUID.randomUUID().toString() + ".npy", rai);
 	        	outputTensors.get(i).setData(Tensor.createCopyOfRaiInWantedDataType(Cast.unchecked(rai), Util.getTypeFromInterval(Cast.unchecked(rai))));
 	        }
 		} catch (Exception e) {

src/main/java/io/bioimage/modelrunner/tensorflow/v1/shm/ShmBuilder.java

@@ -24,12 +24,9 @@
 import io.bioimage.modelrunner.tensor.shm.SharedMemoryArray;
 import io.bioimage.modelrunner.utils.CommonUtils;
 
-import java.io.FileOutputStream;
 import java.io.IOException;
 import java.nio.ByteBuffer;
-import java.nio.channels.FileChannel;
 import java.util.Arrays;
-import java.util.UUID;
 
 import org.tensorflow.Tensor;
 import org.tensorflow.types.UInt8;
@@ -124,18 +121,6 @@ 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()) {
-            	ByteBuffer buffer = shma.getDataBuffer();
-	            // Write the buffer's content to the file
-	            while (buffer.hasRemaining()) {
-	                fileChannel.write(buffer);
-	            }
-	            buffer.rewind();
-		} catch (IOException e) {
-			// TODO Auto-generated catch block
-			e.printStackTrace();
-		}
         if (PlatformDetection.isWindows()) shma.close();
     }