From d8ea1f9f7e97df4d03917cb3dd133ca1fc7b0fe9 Mon Sep 17 00:00:00 2001 From: carlosuc3m <100329787@alumnos.uc3m.es> Date: Tue, 14 Jan 2025 17:05:17 +0100 Subject: [PATCH] preprocess the inputs --- src/main/java/io/bioimage/modelrunner/model/Stardist3D.java | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/main/java/io/bioimage/modelrunner/model/Stardist3D.java b/src/main/java/io/bioimage/modelrunner/model/Stardist3D.java index 6813d0ca..c1e5675f 100644 --- a/src/main/java/io/bioimage/modelrunner/model/Stardist3D.java +++ b/src/main/java/io/bioimage/modelrunner/model/Stardist3D.java @@ -48,6 +48,7 @@ import io.bioimage.modelrunner.exceptions.LoadEngineException; import io.bioimage.modelrunner.exceptions.LoadModelException; import io.bioimage.modelrunner.exceptions.RunModelException; +import io.bioimage.modelrunner.model.processing.Processing; import io.bioimage.modelrunner.runmode.RunMode; import io.bioimage.modelrunner.runmode.ops.GenericOp; import io.bioimage.modelrunner.tensor.Tensor; @@ -244,6 +245,8 @@ RandomAccessibleInterval predict(RandomAccessibleInterval image) throws Mo Model model = Model.createBioimageioModel(this.descriptor.getModelPath()); model.loadModel(); + Processing processing = Processing.init(descriptor); + inputList = processing.preprocess(inputList, false); model.runModel(inputList, outputList); return Utils.transpose(Cast.unchecked(postProcessing(outputList.get(0).getData())));