Fix input_image_input_size

bgruening · Nov 7, 2024 · c51e7b0 · c51e7b0
1 parent 17bb168
commit c51e7b0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 7 deletions.
diff --git a/tools/bioimaging/bioimage_inference.xml b/tools/bioimaging/bioimage_inference.xml
@@ -59,7 +59,7 @@
         <test>
             <param name="input_imaging_model" value="input_imaging_model.zip" location="https://zenodo.org/api/records/7768142/files/torchscript_tracing.pt/content"/>
             <param name="input_image_file" value="input_bheu2024a.tif"/>
-            <param name="input_image_input_size" value="256,256,2,1"/>
+            <param name="input_image_input_size" value="128,128,100,1"/>
             <output name="output_predicted_image" file="output_bheu2024.tif" compare="sim_size" delta="100" />
             <output name="output_predicted_image_matrix" file="output_bheu2024.npy" compare="sim_size" delta="100" />
         </test>

diff --git a/tools/bioimaging/main.py b/tools/bioimaging/main.py
@@ -11,12 +11,7 @@
 
 def find_dim_order(user_in_shape, input_image):
     """
-    Find the correct order of input image's
-    shape. For a few models, the order of input size
-    mentioned in the RDF.yaml file is reversed compared
-    to the input image's original size. If it is reversed,
-    transpose the image to find correct order of image's
-    dimensions.
+    Find the correct order of input image's shape. For a few models, the order of input size mentioned in the RDF.yaml file is reversed compared to the input image's original size. If it is reversed, transpose the image to find correct order of image's dimensions.
     """
     image_shape = list(input_image.shape)
     # reverse the input shape provided from RDF.yaml file