Skip to content

Commit 87ffcbd

Browse files
committed
improve
1 parent 1246da1 commit 87ffcbd

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

src/main/java/io/bioimage/modelrunner/pytorch/PytorchInterface.java

+2-1
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@
6060
import java.util.LinkedList;
6161
import java.util.List;
6262
import java.util.Map;
63+
import java.util.UUID;
6364

6465
import com.google.gson.Gson;
6566

@@ -275,7 +276,7 @@ protected void runFromShmas(List<String> inputs, List<String> outputs) throws IO
275276
for (String ee : inputs) {
276277
Map<String, Object> decoded = Types.decode(ee);
277278
SharedMemoryArray shma = SharedMemoryArray.read((String) decoded.get(MEM_NAME_KEY));
278-
DecodeNumpy.saveNpy("/home/carlos/git/mm_in.npy", Cast.unchecked(shma.getSharedRAI()));
279+
DecodeNumpy.saveNpy("/home/carlos/git/mm_in" + UUID.randomUUID().toString() + ".npy", Cast.unchecked(shma.getSharedRAI()));
279280
NDArray inT = TensorBuilder.build(shma, manager);
280281
if (PlatformDetection.isWindows()) shma.close();
281282
inputList.add(inT);

src/main/java/io/bioimage/modelrunner/pytorch/shm/ShmBuilder.java

+2-1
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727

2828
import java.io.IOException;
2929
import java.util.Arrays;
30+
import java.util.UUID;
3031

3132
import ai.djl.ndarray.NDArray;
3233
import net.imglib2.type.numeric.integer.IntType;
@@ -116,7 +117,7 @@ private static void buildFromTensorFloat(NDArray tensor, String memoryName) thro
116117

117118
SharedMemoryArray shma = SharedMemoryArray.readOrCreate(memoryName, arrayShape, new FloatType(), false, true);
118119
shma.getDataBufferNoHeader().put(tensor.toByteArray());
119-
DecodeNumpy.saveNpy("/home/carlos/git/mm_out.npy", Cast.unchecked(shma.getSharedRAI()));
120+
DecodeNumpy.saveNpy("/home/carlos/git/mm_out" + UUID.randomUUID().toString() + ".npy", Cast.unchecked(shma.getSharedRAI()));
120121
if (PlatformDetection.isWindows()) shma.close();
121122
}
122123

0 commit comments

Comments
 (0)