Add endoscopic bundle name option (#1110)

* Add option for endoscopic bundle name

Signed-off-by: tangy5 <[email protected]>

* Add option for endoscopic bundle name

Signed-off-by: tangy5 <[email protected]>

* Add option for endoscopic bundle name

Signed-off-by: tangy5 <[email protected]>

* Add option for endoscopic bundle name

Signed-off-by: tangy5 <[email protected]>

Signed-off-by: tangy5 <[email protected]>

Author: tangy5

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

@@ -33,8 +33,8 @@ class InBody(TaskConfig):
     def init(self, name: str, model_dir: str, conf: Dict[str, str], planner: Any, **kwargs):
         super().init(name, model_dir, conf, planner, **kwargs)
 
-        bundle_name = "endoscopic_inbody_classification"
-        bundle_version = "0.2.0"
+        bundle_name = conf.get("bundle_name", "endoscopic_inbody_classification")
+        bundle_version = conf.get("bundle_version", "0.2.0")
         self.bundle_path = os.path.join(self.model_dir, bundle_name)
         if not os.path.exists(self.bundle_path):
             download(name=bundle_name, version=bundle_version, bundle_dir=self.model_dir)

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

@@ -33,8 +33,8 @@ class ToolTracking(TaskConfig):
     def init(self, name: str, model_dir: str, conf: Dict[str, str], planner: Any, **kwargs):
         super().init(name, model_dir, conf, planner, **kwargs)
 
-        bundle_name = "endoscopic_tool_segmentation"
-        bundle_version = "0.2.0"
+        bundle_name = conf.get("bundle_name", "endoscopic_tool_segmentation")
+        bundle_version = conf.get("bundle_version", "0.2.0")
         self.bundle_path = os.path.join(self.model_dir, bundle_name)
         if not os.path.exists(self.bundle_path):
             download(name=bundle_name, version=bundle_version, bundle_dir=self.model_dir)