yuxindu
/

segvol

Image Segmentation

feature-extraction

Model card Files Files and versions Community

yuxin commited on Apr 15, 2024

Commit

410f9a9

•

1 Parent(s): 0672fb5

add model

Files changed (1) hide show

model_segvol_single.py +1 -1

model_segvol_single.py CHANGED Viewed

@@ -136,7 +136,7 @@ class SegVolProcessor():
         self.custom_device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
         self.transform4train = transforms.Compose(
         [
-            transforms.AddChanneld(keys=["image"]),
             DimTranspose(keys=["image", "label"]),
             MinMaxNormalization(),
             transforms.CropForegroundd(keys=["image", "label"], source_key="image"),

         self.custom_device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
         self.transform4train = transforms.Compose(
         [
+            # transforms.AddChanneld(keys=["image"]),
             DimTranspose(keys=["image", "label"]),
             MinMaxNormalization(),
             transforms.CropForegroundd(keys=["image", "label"], source_key="image"),