updates on 3mm training

Signed-off-by: tangy5 <[email protected]>
Project-MONAI · Feb 6, 2023 · 876c158 · 876c158
1 parent e2a2409
commit 876c158
Show file tree

Hide file tree

Showing 6 changed files with 448 additions and 11 deletions.
diff --git a/models/wholeBody_ct_segmentation/configs/inference.json b/models/wholeBody_ct_segmentation/configs/inference.json
@@ -4,8 +4,8 @@
         "$import os"
     ],
     "bundle_root": ".",
-    "output_dir": "$@bundle_root + '/eval'",
-    "dataset_dir": "../datasets/Task09_Spleen",
+    "output_dir": "$@bundle_root + '/eval_v2'",
+    "dataset_dir": "../datasets/totalSegmentator",
     "datalist": "$list(sorted(glob.glob(@dataset_dir + '/imagesTs/*.nii.gz')))",
     "device": "$torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')",
     "network_def": {

diff --git a/models/wholeBody_ct_segmentation/configs/train.json b/models/wholeBody_ct_segmentation/configs/train.json
@@ -10,7 +10,7 @@
     "dataset_dir": "../datasets/totalSegmentator",
     "images": "$list(sorted(glob.glob(@dataset_dir + '/images/*.nii.gz')))",
     "labels": "$list(sorted(glob.glob(@dataset_dir + '/labels/*.nii.gz')))",
-    "val_interval": 5,
+    "val_interval": 20,
     "device": "$torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')",
     "network_def": {
         "_target_": "SegResNet",
@@ -151,17 +151,17 @@
         },
         "dataset": {
             "_target_": "CacheDataset",
-            "data": "$[{'image': i, 'label': l} for i, l in zip(@images[:-200], @labels[:-200])]",
+            "data": "$[{'image': i, 'label': l} for i, l in zip(@images[:-57], @labels[:-57])]",
             "transform": "@train#preprocessing",
-            "cache_rate": 0.5,
-            "num_workers": 8
+            "cache_rate": 0.4,
+            "num_workers": 4
         },
         "dataloader": {
             "_target_": "DataLoader",
             "dataset": "@train#dataset",
             "batch_size": 16,
             "shuffle": true,
-            "num_workers": 8
+            "num_workers": 4
         },
         "inferer": {
             "_target_": "SimpleInferer"
@@ -215,7 +215,7 @@
         },
         "trainer": {
             "_target_": "SupervisedTrainer",
-            "max_epochs": 1000,
+            "max_epochs": 4000,
             "device": "@device",
             "train_data_loader": "@train#dataloader",
             "network": "@network",