Skip to content

Commit 34951f3

Browse files
authored
Update tutorials under reconstruction folder (#1671)
Fixes #1670 ### Description Set "image_only=False" in `LoadImage`. ### Checks <!--- Put an `x` in all the boxes that apply, and remove the not applicable items --> - [ ] Avoid including large-size files in the PR. - [ ] Clean up long text outputs from code cells in the notebook. - [ ] For security purposes, please check the contents and remove any sensitive info such as user names and private key. - [ ] Ensure (1) hyperlinks and markdown anchors are working (2) use relative paths for tutorial repo files (3) put figure and graphs in the `./figure` folder - [ ] Notebook runs automatically `./runner.sh -t <path to .ipynb file>` --------- Signed-off-by: YunLiu <[email protected]>
1 parent 570b19e commit 34951f3

File tree

7 files changed

+9
-7
lines changed

7 files changed

+9
-7
lines changed

detection/generate_transforms.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def generate_detection_train_transform(
7878

7979
train_transforms = Compose(
8080
[
81-
LoadImaged(keys=[image_key], meta_key_postfix="meta_dict"),
81+
LoadImaged(keys=[image_key], image_only=False, meta_key_postfix="meta_dict"),
8282
EnsureChannelFirstd(keys=[image_key]),
8383
EnsureTyped(keys=[image_key, box_key], dtype=torch.float32),
8484
EnsureTyped(keys=[label_key], dtype=torch.long),
@@ -224,7 +224,7 @@ def generate_detection_val_transform(
224224

225225
val_transforms = Compose(
226226
[
227-
LoadImaged(keys=[image_key], meta_key_postfix="meta_dict"),
227+
LoadImaged(keys=[image_key], image_only=False, meta_key_postfix="meta_dict"),
228228
EnsureChannelFirstd(keys=[image_key]),
229229
EnsureTyped(keys=[image_key, box_key], dtype=torch.float32),
230230
EnsureTyped(keys=[label_key], dtype=torch.long),
@@ -280,7 +280,7 @@ def generate_detection_inference_transform(
280280

281281
test_transforms = Compose(
282282
[
283-
LoadImaged(keys=[image_key], meta_key_postfix="meta_dict"),
283+
LoadImaged(keys=[image_key], image_only=False, meta_key_postfix="meta_dict"),
284284
EnsureChannelFirstd(keys=[image_key]),
285285
EnsureTyped(keys=[image_key], dtype=torch.float32),
286286
Orientationd(keys=[image_key], axcodes="RAS"),

detection/luna16_prepare_images.py

+1
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,7 @@ def main():
6161
[
6262
LoadImaged(
6363
keys=["image"],
64+
image_only=False,
6465
meta_key_postfix="meta_dict",
6566
reader="itkreader",
6667
affine_lps_to_ras=True,

detection/luna16_prepare_images_dicom.py

+1
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,7 @@ def main():
6363
[
6464
LoadImaged(
6565
keys=["image"],
66+
image_only=False,
6667
meta_key_postfix="meta_dict",
6768
reader="itkreader",
6869
affine_lps_to_ras=True,

reconstruction/MRI_reconstruction/unet_demo/inference.ipynb

+1-1
Original file line numberDiff line numberDiff line change
@@ -236,7 +236,7 @@
236236
"\n",
237237
"test_transforms = Compose(\n",
238238
" [\n",
239-
" LoadImaged(keys=[\"kspace\"], reader=FastMRIReader, dtype=np.complex64),\n",
239+
" LoadImaged(keys=[\"kspace\"], reader=FastMRIReader, image_only=False, dtype=np.complex64),\n",
240240
" # user can also add other random transforms\n",
241241
" ExtractDataKeyFromMetaKeyd(keys=[\"reconstruction_rss\", \"mask\"], meta_key=\"kspace_meta_dict\"),\n",
242242
" MaskTransform,\n",

reconstruction/MRI_reconstruction/unet_demo/train.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ def trainer(args):
9797

9898
train_transforms = Compose(
9999
[
100-
LoadImaged(keys=["kspace"], reader=FastMRIReader, dtype=np.complex64),
100+
LoadImaged(keys=["kspace"], reader=FastMRIReader, image_only=False, dtype=np.complex64),
101101
# user can also add other random transforms
102102
ExtractDataKeyFromMetaKeyd(keys=["reconstruction_rss", "mask"], meta_key="kspace_meta_dict"),
103103
MaskTransform,

reconstruction/MRI_reconstruction/varnet_demo/inference.ipynb

+1-1
Original file line numberDiff line numberDiff line change
@@ -251,7 +251,7 @@
251251
"\n",
252252
"test_transforms = Compose(\n",
253253
" [\n",
254-
" LoadImaged(keys=[\"kspace\"], reader=FastMRIReader, dtype=np.complex64),\n",
254+
" LoadImaged(keys=[\"kspace\"], reader=FastMRIReader, image_only=False, dtype=np.complex64),\n",
255255
" # user can also add other random transforms\n",
256256
" ExtractDataKeyFromMetaKeyd(keys=[\"reconstruction_rss\", \"mask\"], meta_key=\"kspace_meta_dict\"),\n",
257257
" MaskTransform,\n",

reconstruction/MRI_reconstruction/varnet_demo/train.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ def trainer(args):
106106

107107
train_transforms = Compose(
108108
[
109-
LoadImaged(keys=["kspace"], reader=FastMRIReader, dtype=np.complex64),
109+
LoadImaged(keys=["kspace"], reader=FastMRIReader, image_only=False, dtype=np.complex64),
110110
# user can also add other random transforms but remember to disable randomness for val_transforms
111111
ExtractDataKeyFromMetaKeyd(keys=["reconstruction_rss", "mask"], meta_key="kspace_meta_dict"),
112112
MaskTransform,

0 commit comments

Comments
 (0)