Mercurial > repos > goeckslab > image_learner

changeset 3:2c3a3dfaf1a9 draft default tip

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload for repository https://github.com/goeckslab/gleam.git commit 5ab02b6688c9af85525dd225134db8f2bbed76ed
author goeckslab
date Fri, 04 Jul 2025 03:44:56 +0000 (2 days ago)
parents 186424a7eca7
children
files constants.py image_learner.xml
diffstat 2 files changed, 28 insertions(+), 8 deletions(-) [+]
line wrap: on
line diff
--- a/constants.py	Thu Jul 03 20:43:24 2025 +0000
+++ b/constants.py	Fri Jul 04 03:44:56 2025 +0000
@@ -68,11 +68,11 @@
     "swin_v2_t": {"type": "swin_transformer", "model_variant": "v2_t"},
     "swin_v2_s": {"type": "swin_transformer", "model_variant": "v2_s"},
     "swin_v2_b": {"type": "swin_transformer", "model_variant": "v2_b"},
-    "vit_b_16": {"type": "vision_transformer", "model_variant": "b_16"},
-    "vit_b_32": {"type": "vision_transformer", "model_variant": "b_32"},
-    "vit_l_16": {"type": "vision_transformer", "model_variant": "l_16"},
-    "vit_l_32": {"type": "vision_transformer", "model_variant": "l_32"},
-    "vit_h_14": {"type": "vision_transformer", "model_variant": "h_14"},
+    "vit_b_16": {"type": "vit", "model_variant": "b_16"},
+    "vit_b_32": {"type": "vit", "model_variant": "b_32"},
+    "vit_l_16": {"type": "vit", "model_variant": "l_16"},
+    "vit_l_32": {"type": "vit", "model_variant": "l_32"},
+    "vit_h_14": {"type": "vit", "model_variant": "h_14"},
     "convnext_tiny": {"type": "convnext", "model_variant": "tiny"},
     "convnext_small": {"type": "convnext", "model_variant": "small"},
     "convnext_base": {"type": "convnext", "model_variant": "base"},
@@ -80,10 +80,10 @@
     "maxvit_t": {"type": "maxvit", "model_variant": "t"},
     "alexnet": {"type": "alexnet"},
     "googlenet": {"type": "googlenet"},
-    "inception_v3": {"type": "inception_v3"},
+    "inception_v3": {"type": "inceptionv3"},
     "mobilenet_v2": {"type": "mobilenet_v2"},
-    "mobilenet_v3_large": {"type": "mobilenet_v3_large"},
-    "mobilenet_v3_small": {"type": "mobilenet_v3_small"},
+    "mobilenet_v3_large": {"type": "mobilenetv3", "model_variant": "large"},
+    "mobilenet_v3_small": {"type": "mobilenetv3", "model_variant": "small"},
 }
 METRIC_DISPLAY_NAMES = {
     "accuracy": "Accuracy",
--- a/image_learner.xml	Thu Jul 03 20:43:24 2025 +0000
+++ b/image_learner.xml	Fri Jul 04 03:44:56 2025 +0000
@@ -254,6 +254,26 @@
                 </element>
             </output_collection>
         </test>
+         <test expect_num_outputs="3">
+            <param name="input_csv" value="mnist_subset.csv" ftype="csv" />
+            <param name="image_zip" value="mnist_subset.zip" ftype="zip" />
+            <param name="model_name" value="vit_b_16" />
+            <output name="output_report">
+                <assert_contents>
+                    <has_text text="Results Summary" />
+                    <has_text text="Train/Validation Results" />
+                    <has_text text="Test Results" />
+                </assert_contents>
+            </output>
+
+            <output_collection name="output_pred_csv" type="list" >
+                <element name="predictions.csv" >
+                    <assert_contents>
+                        <has_n_columns n="1" />
+                    </assert_contents>
+                </element>
+            </output_collection>
+        </test>
         <test expect_num_outputs="3">
             <param name="input_csv" value="mnist_subset.csv" ftype="csv" />
             <param name="image_zip" value="mnist_subset.zip" ftype="zip" />