10
10
from .mitochondria import segment_mitochondria
11
11
from .ribbon_synapse import segment_ribbon_synapse_structures
12
12
from .vesicles import segment_vesicles
13
+ from .cristae import segment_cristae
13
14
from .util import get_device
14
15
from ..file_utils import get_cache_dir
15
16
@@ -25,6 +26,7 @@ def _get_model_registry():
25
26
"compartments" : "527983720f9eb215c45c4f4493851fd6551810361eda7b79f185a0d304274ee1" ,
26
27
"mitochondria" : "24625018a5968b36f39fa9d73b121a32e8f66d0f2c0540d3df2e1e39b3d58186" ,
27
28
"mitochondria2" : "553decafaff4838fff6cc8347f22c8db3dee5bcbeffc34ffaec152f8449af673" ,
29
+ "cristae" : "f96c90484f4ea92ac0515a06e389cc117580f02c2aacdc44b5828820cf38c3c3" ,
28
30
"ribbon" : "7c947f0ddfabe51a41d9d05c0a6ca7d6b238f43df2af8fffed5552d09bb075a9" ,
29
31
"vesicles_2d" : "eb0b74f7000a0e6a25b626078e76a9452019f2d1ea6cf2033073656f4f055df1" ,
30
32
"vesicles_3d" : "b329ec1f57f305099c984fbb3d7f6ae4b0ff51ec2fa0fa586df52dad6b84cf29" ,
@@ -35,6 +37,7 @@ def _get_model_registry():
35
37
"compartments" : "https://owncloud.gwdg.de/index.php/s/DnFDeTmDDmZrDDX/download" ,
36
38
"mitochondria" : "https://owncloud.gwdg.de/index.php/s/1T542uvzfuruahD/download" ,
37
39
"mitochondria2" : "https://owncloud.gwdg.de/index.php/s/GZghrXagc54FFXd/download" ,
40
+ "cristae" : "https://owncloud.gwdg.de/index.php/s/Df7OUOyQ1Kc2eEO/download" ,
38
41
"ribbon" : "https://owncloud.gwdg.de/index.php/s/S3b5l0liPP1XPYA/download" ,
39
42
"vesicles_2d" : "https://owncloud.gwdg.de/index.php/s/d72QIvdX6LsgXip/download" ,
40
43
"vesicles_3d" : "https://owncloud.gwdg.de/index.php/s/A425mkAOSqePDhx/download" ,
@@ -214,14 +217,16 @@ def run_segmentation(
214
217
"""
215
218
if model_type .startswith ("vesicles" ):
216
219
segmentation = segment_vesicles (image , model = model , tiling = tiling , scale = scale , verbose = verbose , ** kwargs )
217
- elif model_type == "mitochondria" :
220
+ elif model_type == "mitochondria" or model_type == "mitochondria2" :
218
221
segmentation = segment_mitochondria (image , model = model , tiling = tiling , scale = scale , verbose = verbose , ** kwargs )
219
222
elif model_type == "active_zone" :
220
223
segmentation = segment_active_zone (image , model = model , tiling = tiling , scale = scale , verbose = verbose , ** kwargs )
221
224
elif model_type == "compartments" :
222
225
segmentation = segment_compartments (image , model = model , tiling = tiling , scale = scale , verbose = verbose , ** kwargs )
223
226
elif model_type == "ribbon" :
224
227
segmentation = _segment_ribbon_AZ (image , model = model , tiling = tiling , scale = scale , verbose = verbose , ** kwargs )
228
+ elif model_type == "cristae" :
229
+ segmentation = segment_cristae (image , model = model , tiling = tiling , scale = scale , verbose = verbose , ** kwargs )
225
230
else :
226
231
raise ValueError (f"Unknown model type: { model_type } " )
227
232
return segmentation
0 commit comments