Mercurial > repos > goeckslab > image_learner
comparison constants.py @ 3:2c3a3dfaf1a9 draft default tip
planemo upload for repository https://github.com/goeckslab/gleam.git commit 5ab02b6688c9af85525dd225134db8f2bbed76ed
author | goeckslab |
---|---|
date | Fri, 04 Jul 2025 03:44:56 +0000 |
parents | 186424a7eca7 |
children |
comparison
equal
deleted
inserted
replaced
2:186424a7eca7 | 3:2c3a3dfaf1a9 |
---|---|
66 "swin_s": {"type": "swin_transformer", "model_variant": "s"}, | 66 "swin_s": {"type": "swin_transformer", "model_variant": "s"}, |
67 "swin_b": {"type": "swin_transformer", "model_variant": "b"}, | 67 "swin_b": {"type": "swin_transformer", "model_variant": "b"}, |
68 "swin_v2_t": {"type": "swin_transformer", "model_variant": "v2_t"}, | 68 "swin_v2_t": {"type": "swin_transformer", "model_variant": "v2_t"}, |
69 "swin_v2_s": {"type": "swin_transformer", "model_variant": "v2_s"}, | 69 "swin_v2_s": {"type": "swin_transformer", "model_variant": "v2_s"}, |
70 "swin_v2_b": {"type": "swin_transformer", "model_variant": "v2_b"}, | 70 "swin_v2_b": {"type": "swin_transformer", "model_variant": "v2_b"}, |
71 "vit_b_16": {"type": "vision_transformer", "model_variant": "b_16"}, | 71 "vit_b_16": {"type": "vit", "model_variant": "b_16"}, |
72 "vit_b_32": {"type": "vision_transformer", "model_variant": "b_32"}, | 72 "vit_b_32": {"type": "vit", "model_variant": "b_32"}, |
73 "vit_l_16": {"type": "vision_transformer", "model_variant": "l_16"}, | 73 "vit_l_16": {"type": "vit", "model_variant": "l_16"}, |
74 "vit_l_32": {"type": "vision_transformer", "model_variant": "l_32"}, | 74 "vit_l_32": {"type": "vit", "model_variant": "l_32"}, |
75 "vit_h_14": {"type": "vision_transformer", "model_variant": "h_14"}, | 75 "vit_h_14": {"type": "vit", "model_variant": "h_14"}, |
76 "convnext_tiny": {"type": "convnext", "model_variant": "tiny"}, | 76 "convnext_tiny": {"type": "convnext", "model_variant": "tiny"}, |
77 "convnext_small": {"type": "convnext", "model_variant": "small"}, | 77 "convnext_small": {"type": "convnext", "model_variant": "small"}, |
78 "convnext_base": {"type": "convnext", "model_variant": "base"}, | 78 "convnext_base": {"type": "convnext", "model_variant": "base"}, |
79 "convnext_large": {"type": "convnext", "model_variant": "large"}, | 79 "convnext_large": {"type": "convnext", "model_variant": "large"}, |
80 "maxvit_t": {"type": "maxvit", "model_variant": "t"}, | 80 "maxvit_t": {"type": "maxvit", "model_variant": "t"}, |
81 "alexnet": {"type": "alexnet"}, | 81 "alexnet": {"type": "alexnet"}, |
82 "googlenet": {"type": "googlenet"}, | 82 "googlenet": {"type": "googlenet"}, |
83 "inception_v3": {"type": "inception_v3"}, | 83 "inception_v3": {"type": "inceptionv3"}, |
84 "mobilenet_v2": {"type": "mobilenet_v2"}, | 84 "mobilenet_v2": {"type": "mobilenet_v2"}, |
85 "mobilenet_v3_large": {"type": "mobilenet_v3_large"}, | 85 "mobilenet_v3_large": {"type": "mobilenetv3", "model_variant": "large"}, |
86 "mobilenet_v3_small": {"type": "mobilenet_v3_small"}, | 86 "mobilenet_v3_small": {"type": "mobilenetv3", "model_variant": "small"}, |
87 } | 87 } |
88 METRIC_DISPLAY_NAMES = { | 88 METRIC_DISPLAY_NAMES = { |
89 "accuracy": "Accuracy", | 89 "accuracy": "Accuracy", |
90 "accuracy_micro": "Accuracy-Micro", | 90 "accuracy_micro": "Accuracy-Micro", |
91 "loss": "Loss", | 91 "loss": "Loss", |