updated downloadable model default path

danilexn · danilexn · commit 69eefc9714bf · 2024-04-19T09:51:20.000+02:00
diff --git a/openst/preprocessing/image_preprocess.py b/openst/preprocessing/image_preprocess.py
@@ -25,7 +25,7 @@
     "HE_CUT_rajewsky"
 ]
 
-MODEL_DIR = pathlib.Path.home().joinpath(".CUT", "models")
+MODEL_DIR = pathlib.Path.home().joinpath(".openst", "CUT", "models")
 _MODEL_URL = "http://bimsbstatic.mdc-berlin.de/rajewsky/openst-public-data/CUT_models"
 
 def create_dataset(args, opt):
diff --git a/openst/segmentation/segment.py b/openst/segmentation/segment.py
@@ -32,7 +32,7 @@
 ]
 
 # adapted from cellpose
-MODEL_DIR = pathlib.Path.home().joinpath(".cellpose", "models")
+MODEL_DIR = pathlib.Path.home().joinpath(".openst", "cellpose", "models")
 _MODEL_URL = "http://bimsbstatic.mdc-berlin.de/rajewsky/openst-public-data/models"
 
 def cache_model_path(basename):

Original file line number	Diff line number	Diff line change
`@@ -25,7 +25,7 @@`
`25`	`25`	`"HE_CUT_rajewsky"`
`26`	`26`	`]`
`27`	`27`
`28`		`-MODEL_DIR = pathlib.Path.home().joinpath(".CUT", "models")`
	`28`	`+MODEL_DIR = pathlib.Path.home().joinpath(".openst", "CUT", "models")`
`29`	`29`	`_MODEL_URL = "http://bimsbstatic.mdc-berlin.de/rajewsky/openst-public-data/CUT_models"`
`30`	`30`
`31`	`31`	`def create_dataset(args, opt):`
Original file line number	Diff line number	Diff line change
`@@ -32,7 +32,7 @@`
`32`	`32`	`]`
`33`	`33`
`34`	`34`	`# adapted from cellpose`
`35`		`-MODEL_DIR = pathlib.Path.home().joinpath(".cellpose", "models")`
	`35`	`+MODEL_DIR = pathlib.Path.home().joinpath(".openst", "cellpose", "models")`
`36`	`36`	`_MODEL_URL = "http://bimsbstatic.mdc-berlin.de/rajewsky/openst-public-data/models"`
`37`	`37`
`38`	`38`	`def cache_model_path(basename):`