Use test inputs if no sample inputs exists

bioimage-io · Sep 28, 2023 · 9463789 · 9463789
1 parent 0f786ee
commit 9463789
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/public/plugins/bioengine-test-run.imjoy.html b/public/plugins/bioengine-test-run.imjoy.html
@@ -257,6 +257,10 @@
             const sampleInput = rdf.sample_inputs[0]
             this.set_default_url(sampleInput)
         }
+        else{
+            const testInputs = rdf.test_inputs[0]
+            this.set_default_url(testInputs)
+        }
         this.setInfoPanel("")
         this.setWaiting(false)
         this.setButtonsDisabled(false)
@@ -553,6 +557,9 @@
         _file.name = file_name
         if file_name.endswith(".tif") or file_name.endswith(".tiff"):
             image = imageio.volread(_file)
+        if file_name.endswith(".npy"):
+            image = np.load(_file)
+            image = image.squeeze()
         else:
             image = imageio.imread(_file)
         await api.log(