update deep_atlas (#823)

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

Author: KumoLiu

deep_atlas/deep_atlas_tutorial.ipynb

@@ -503,9 +503,8 @@
     "transform_seg_available = monai.transforms.Compose(\n",
     "    transforms=[\n",
     "        monai.transforms.LoadImageD(keys=['img', 'seg'], image_only=True),\n",
-    "        monai.transforms.ToTensorD(keys=['img', 'seg']),\n",
     "        monai.transforms.TransposeD(keys=['img', 'seg'], indices=(2, 1, 0)),\n",
-    "        monai.transforms.AddChannelD(keys=['img', 'seg']),\n",
+    "        monai.transforms.EnsureChannelFirstD(keys=['img', 'seg']),\n",
     "        monai.transforms.ResizeD(\n",
     "            keys=['img', 'seg'],\n",
     "            spatial_size=(resize, resize, resize),\n",
@@ -740,9 +739,8 @@
     "transform_pair = monai.transforms.Compose(\n",
     "    transforms=[\n",
     "        monai.transforms.LoadImageD(keys=['img1', 'seg1', 'img2', 'seg2'], image_only=True, allow_missing_keys=True),\n",
-    "        monai.transforms.ToTensorD(keys=['img1', 'seg1', 'img2', 'seg2'], allow_missing_keys=True),\n",
     "        monai.transforms.TransposeD(keys=['img1', 'seg1', 'img2', 'seg2'], indices=(2, 1, 0), allow_missing_keys=True),\n",
-    "        monai.transforms.AddChannelD(keys=['img1', 'seg1', 'img2', 'seg2'], allow_missing_keys=True),\n",
+    "        monai.transforms.EnsureChannelFirstD(keys=['img1', 'seg1', 'img2', 'seg2'], allow_missing_keys=True),\n",
     "        monai.transforms.ConcatItemsD(keys=['img1', 'img2'], name='img12', dim=0),\n",
     "        monai.transforms.DeleteItemsD(keys=['img1', 'img2']),\n",
     "        monai.transforms.ResizeD(\n",
@@ -889,7 +887,7 @@
     "\n",
     "data_item = random.choice(dataset_seg_available_train)\n",
     "seg_net_example_output = seg_net(data_item['img'].unsqueeze(0))\n",
-    "print(f\"Segmentation classes: {torch.unique(data_item['seg'])}\")\n",
+    "print(f\"Segmentation classes: {torch.unique(data_item['seg']).as_tensor()}\")\n",
     "print(f\"Shape of ground truth label: {data_item['seg'].unsqueeze(0).shape}\")\n",
     "print(f\"Shape of seg_net output: {seg_net_example_output.shape}\")"
    ]
@@ -3037,7 +3035,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.10"
+   "version": "3.8.13"
   }
  },
  "nbformat": 4,