Skip to content

Commit 81f4037

Browse files
committed
Adding type conversion fix
Signed-off-by: Eric Kerfoot <[email protected]>
1 parent 81530e0 commit 81f4037

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

model_zoo/TCIA_PROSTATEx_Prostate_MRI_Anatomy_Model.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -593,7 +593,7 @@
593593
"# for inference. As a result, the result image may not be in the same\n",
594594
"# spacing, orientation, etc as the original input data. So, we resample the results\n",
595595
"# image to match the physical properties of the original input data.\n",
596-
"interpolator = itk.NearestNeighborInterpolateImageFunction.New(seg_image)\n",
596+
"interpolator = itk.NearestNeighborInterpolateImageFunction.New(seg_image.astype(result_image.dtype))\n",
597597
"result_image_resampled = itk.resample_image_filter(\n",
598598
" Input=result_image, Interpolator=interpolator, reference_image=seg_image_prep, use_reference_image=True\n",
599599
")"

0 commit comments

Comments
 (0)