File tree 1 file changed +2
-0
lines changed
src/main/java/io/bioimage/modelrunner/pytorch
1 file changed +2
-0
lines changed Original file line number Diff line number Diff line change 27
27
import io .bioimage .modelrunner .engine .DeepLearningEngineInterface ;
28
28
import io .bioimage .modelrunner .exceptions .LoadModelException ;
29
29
import io .bioimage .modelrunner .exceptions .RunModelException ;
30
+ import io .bioimage .modelrunner .numpy .DecodeNumpy ;
30
31
import io .bioimage .modelrunner .pytorch .shm .ShmBuilder ;
31
32
import io .bioimage .modelrunner .pytorch .shm .TensorBuilder ;
32
33
import io .bioimage .modelrunner .pytorch .tensor .ImgLib2Builder ;
@@ -338,6 +339,7 @@ else if (task.status == TaskStatus.CRASHED) {
338
339
shmaOutputList .add (shm );
339
340
}
340
341
RandomAccessibleInterval <?> rai = shm .getSharedRAI ();
342
+ DecodeNumpy .saveNpy ("/home/carlos/git/mm.npy" , Cast .unchecked (rai ));
341
343
outputTensors .get (i ).setData (Tensor .createCopyOfRaiInWantedDataType (Cast .unchecked (rai ), Util .getTypeFromInterval (Cast .unchecked (rai ))));
342
344
}
343
345
} catch (Exception e ) {
You can’t perform that action at this time.
0 commit comments