Skip to content

Commit 8c8be19

Browse files
committed
correct small detail sending shm name
1 parent dba3d46 commit 8c8be19

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

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

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -438,10 +438,9 @@ private List<String> encodeOutputs(List<Tensor<?>> outputTensors) throws FileAlr
438438
shmaOutputList.add(shma);
439439
map.put(MEM_NAME_KEY, shma.getName());
440440
} else if (PlatformDetection.isWindows()){
441-
String memName = SharedMemoryArray.createShmName();
442441
SharedMemoryArray shma = SharedMemoryArray.create(0);
443442
shmaOutputList.add(shma);
444-
map.put(MEM_NAME_KEY, memName);
443+
map.put(MEM_NAME_KEY, shma.getName());
445444
} else {
446445
String memName = SharedMemoryArray.createShmName();
447446
map.put(MEM_NAME_KEY, memName);
@@ -573,13 +572,15 @@ private static String padSpecialJavaBin(String javaBin) {
573572
public static void main(String[] args) throws LoadModelException, IOException, RunModelException {
574573
if (args.length == 0) {
575574

576-
String modelFolder = "/home/carlos/git/deep-icy/models/DeepSupervisionUnet_07112023_172430";
577-
String modelSourc = modelFolder + "/dsUnet.pt";
575+
String modelFolder = "C:\\Users\\carlos\\OneDrive\\Documentos\\pasteur\\git"
576+
+ "\\deep-icy\\models\\MitchondriaEMSegmentation2D_31102023_214027";
577+
String modelSourc = modelFolder + "\\weights-torchscript.pt";
578578
PytorchInterface pi = new PytorchInterface();
579579
pi.loadModel(modelFolder, modelSourc);
580-
RandomAccessibleInterval<FloatType> rai = ArrayImgs.floats(new long[] {1, 3, 1024, 1024});
580+
RandomAccessibleInterval<FloatType> rai = ArrayImgs.floats(new long[] {1, 1, 512, 512});
581581
Tensor<?> inp = Tensor.build("aa", "bcyx", rai);
582-
Tensor<?> out = Tensor.buildBlankTensor("oo", "bcyx", new long[] {1, 9, 1024, 1024}, new FloatType());
582+
Tensor<?> out = Tensor.buildBlankTensor("oo", "bcyx", new long[] {1, 2, 512, 512}, new FloatType());
583+
//Tensor<?> out = Tensor.buildEmptyTensor("oo", "bcyx");
583584
List<Tensor<?>> ins = new ArrayList<Tensor<?>>();
584585
List<Tensor<?>> ous = new ArrayList<Tensor<?>>();
585586
ins.add(inp);

0 commit comments

Comments
 (0)