fix download error

Signed-off-by: Sachidanand Alle <[email protected]>

Author: SachidanandAlle

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

@@ -45,7 +45,10 @@ def init(self, name: str, model_dir: str, conf: Dict[str, str], planner: Any, **
         # Download PreTrained Model
         if strtobool(self.conf.get("use_pretrained_model", "true")):
             url = f"{self.conf.get('pretrained_path', self.PRE_TRAINED_PATH)}/endoscopy_deepedit_tooltracking.pt"
-            download_file(url, self.path[0])
+            try:
+                download_file(url, self.path[0])
+            except:
+                logger.warning(f"Failed to download pre-trained model from {url}; Ignoring the same...")
 
         # Network
         f = (32, 64, 128, 256, 512, 32)

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

@@ -45,7 +45,10 @@ def init(self, name: str, model_dir: str, conf: Dict[str, str], planner: Any, **
         # Download PreTrained Model
         if strtobool(self.conf.get("use_pretrained_model", "true")):
             url = f"{self.conf.get('pretrained_path', self.PRE_TRAINED_PATH)}/endoscopy_deid.pt"
-            download_file(url, self.path[0])
+            try:
+                download_file(url, self.path[0])
+            except:
+                logger.warning(f"Failed to download pre-trained model from {url}; Ignoring the same...")
 
         # Network
         self.network = SEResNet50(spatial_dims=2, in_channels=3, num_classes=2)

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

@@ -46,7 +46,10 @@ def init(self, name: str, model_dir: str, conf: Dict[str, str], planner: Any, **
         # Download PreTrained Model
         if strtobool(self.conf.get("use_pretrained_model", "true")):
             url = f"{self.conf.get('pretrained_path', self.PRE_TRAINED_PATH)}/endoscopy_tooltracking.pt"
-            download_file(url, self.path[0])
+            try:
+                download_file(url, self.path[0])
+            except:
+                logger.warning(f"Failed to download pre-trained model from {url}; Ignoring the same...")
 
         # Network
         self.network = FlexibleUNet(

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

@@ -79,8 +79,7 @@ def train_post_transforms(self, context: Context):
             AsDiscreted(
                 keys=("pred", "label"),
                 argmax=(True, False),
-                to_onehot=True,
-                n_classes=2,
+                to_onehot=(2, 2),
             ),
         ]