Skip to content

Commit 40fd783

Browse files
support endoscopy bundle over windows (#1224)
Signed-off-by: Sachidanand Alle <[email protected]> Signed-off-by: Sachidanand Alle <[email protected]>
1 parent e9b9aa1 commit 40fd783

File tree

2 files changed

+12
-4
lines changed

2 files changed

+12
-4
lines changed

sample-apps/endoscopy/lib/configs/inbody.py

+6-2
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
import lib.infers
1717
import lib.trainers
1818
from lib.scoring.cvat import CVATRandomScoring
19-
from monai.bundle import download
19+
from monai.bundle import download, get_bundle_versions
2020

2121
from monailabel.interfaces.config import TaskConfig
2222
from monailabel.interfaces.tasks.infer_v2 import InferTask
@@ -34,7 +34,11 @@ def init(self, name: str, model_dir: str, conf: Dict[str, str], planner: Any, **
3434
super().init(name, model_dir, conf, planner, **kwargs)
3535

3636
bundle_name = "endoscopic_inbody_classification"
37-
bundle_version = None
37+
repo_owner, repo_name, tag_name = "Project-MONAI/model-zoo/hosting_storage_v1".split("/")
38+
bundle_version = get_bundle_versions(bundle_name, repo=f"{repo_owner}/{repo_name}", tag=tag_name)[
39+
"latest_version"
40+
]
41+
3842
self.bundle_path = os.path.join(self.model_dir, bundle_name)
3943
if not os.path.exists(self.bundle_path):
4044
download(name=bundle_name, version=bundle_version, bundle_dir=self.model_dir)

sample-apps/endoscopy/lib/configs/tooltracking.py

+6-2
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
import lib.infers
1717
import lib.trainers
1818
from lib.scoring.cvat import CVATEpistemicScoring
19-
from monai.bundle import download
19+
from monai.bundle import download, get_bundle_versions
2020

2121
from monailabel.interfaces.config import TaskConfig
2222
from monailabel.interfaces.tasks.infer_v2 import InferTask
@@ -34,7 +34,11 @@ def init(self, name: str, model_dir: str, conf: Dict[str, str], planner: Any, **
3434
super().init(name, model_dir, conf, planner, **kwargs)
3535

3636
bundle_name = "endoscopic_tool_segmentation"
37-
bundle_version = None
37+
repo_owner, repo_name, tag_name = "Project-MONAI/model-zoo/hosting_storage_v1".split("/")
38+
bundle_version = get_bundle_versions(bundle_name, repo=f"{repo_owner}/{repo_name}", tag=tag_name)[
39+
"latest_version"
40+
]
41+
3842
self.bundle_path = os.path.join(self.model_dir, bundle_name)
3943
if not os.path.exists(self.bundle_path):
4044
download(name=bundle_name, version=bundle_version, bundle_dir=self.model_dir)

0 commit comments

Comments
 (0)