add method

bioimage-io · Feb 17, 2025 · 8134d1d · 8134d1d
1 parent 6bcd5d8
commit 8134d1d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/main/java/io/bioimage/modelrunner/model/stardist/StardistAbstract.java b/src/main/java/io/bioimage/modelrunner/model/stardist/StardistAbstract.java
@@ -327,6 +327,7 @@ public void close() {
 	void run( List< Tensor < T > > inTensors, List< Tensor < R > > outTensors ) throws RunModelException {
 		if (inTensors.size() > 1)
 			throw new RunModelException("Stardist needs just one input image");
+		preprocess(inTensors);
 		try {
 			Map<String, RandomAccessibleInterval<R>> outputs = run(inTensors.get(0).getData());
 			for (Tensor<R> tensor : outTensors) {
@@ -374,6 +375,7 @@ List<Tensor<T>> run(List<Tensor<R>> inputTensors)
 			throws RunModelException {
 		if (inputTensors.size() > 1)
 			throw new RunModelException("Stardist needs just one input image");
+		preprocess(inputTensors);
 		try {
 			Map<String, RandomAccessibleInterval<T>> outputs = run(inputTensors.get(0).getData());
 			List<Tensor<T>> outTensors = new ArrayList<Tensor<T>>();