Merge pull request #1103 from computational-cell-analytics/add-new-mi-models

Release new medical imaging models

Author: constantinpape

micro_sam/sam_annotator/_state.py

@@ -116,7 +116,7 @@ def progress_bar_factory(model_type):
                 checkpoint_path=checkpoint_path, return_state=True,
                 progress_bar_factory=None if use_cli else progress_bar_factory,
             )
-            if prefer_decoder and "decoder_state" in state:
+            if prefer_decoder and "decoder_state" in state and model_type != "vit_b_medical_imaging":
                 self.decoder = get_decoder(
                     image_encoder=self.predictor.model.image_encoder,
                     decoder_state=state["decoder_state"],

micro_sam/util.py

@@ -117,7 +117,7 @@ def models():
         "vit_l_histopathology": "xxh128:b591833c89754271023e901281dee3f2",
         "vit_h_histopathology": "xxh128:bd1856dafc156a43fb3aa705f1a6e92e",
         # Medical Imaging models:
-        "vit_b_medical_imaging": "xxh128:5be672f1458263a9edc9fd40d7f56ac1",
+        "vit_b_medical_imaging": "xxh128:40169f1e3c03a4b67bff58249c176d92",
     }
     # Additional decoders for instance segmentation.
     decoder_registry = {
@@ -133,6 +133,8 @@ def models():
         "vit_b_histopathology_decoder": "xxh128:6a66194dcb6e36199cbee2214ecf7213",
         "vit_l_histopathology_decoder": "xxh128:46aab7765d4400e039772d5a50b55c04",
         "vit_h_histopathology_decoder": "xxh128:3ed9f87e46ad5e16935bd8d722c8dc47",
+        # Medical Imaging models:
+        "vit_b_medical_imaging_decoder": "xxh128:9e498b12f526f119b96c88be76e3b2ed",
     }
     registry = {**encoder_registry, **decoder_registry}
 
@@ -150,7 +152,7 @@ def models():
         "vit_b_histopathology": "https://owncloud.gwdg.de/index.php/s/sBB4H8CTmIoBZsQ/download",
         "vit_l_histopathology": "https://owncloud.gwdg.de/index.php/s/IZgnn1cpBq2PHod/download",
         "vit_h_histopathology": "https://owncloud.gwdg.de/index.php/s/L7AcvVz7DoWJ2RZ/download",
-        "vit_b_medical_imaging": "https://owncloud.gwdg.de/index.php/s/AB69HGhj8wuozXQ/download",
+        "vit_b_medical_imaging": "https://owncloud.gwdg.de/index.php/s/f5Ol4FrjPQWfjUF/download",
     }
 
     decoder_urls = {
@@ -163,6 +165,7 @@ def models():
         "vit_b_histopathology_decoder": "https://owncloud.gwdg.de/index.php/s/KO9AWqynI7SFOBj/download",
         "vit_l_histopathology_decoder": "https://owncloud.gwdg.de/index.php/s/oIs6VSmkOp7XrKF/download",
         "vit_h_histopathology_decoder": "https://owncloud.gwdg.de/index.php/s/1qAKxy5H0jgwZvM/download",
+        "vit_b_medical_imaging_decoder": "https://owncloud.gwdg.de/index.php/s/ahd3ZhZl2e0RIwz/download",
     }
     urls = {**encoder_urls, **decoder_urls}