Fix missing func of load_dcm

zhwesky2010 · Nov 7, 2022 · b3f8c06 · b3f8c06
1 parent ae6678f
commit b3f8c06
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/...ased Solution for RSNA Intracranial Hemorrhage Detection/easymia/transforms/functional.py b/...ased Solution for RSNA Intracranial Hemorrhage Detection/easymia/transforms/functional.py
@@ -177,3 +177,16 @@ def hu2uint8(image, HU_min=-1200.0, HU_max=600.0, HU_nan=-2000.0):
     image_new = (image_new * 255).astype('uint8')
 
     return image_new
+
+
+def load_dcm(sorted_dcm_list):
+    """
+    Return img array and [z,y,x]-ordered origin and spacing
+    """
+    itkimage = sitk.ReadImage(sorted_dcm_list)
+    numpyImage = sitk.GetArrayFromImage(itkimage)
+
+    numpyOrigin = np.array(list(reversed(itkimage.GetOrigin())))
+    numpySpacing = np.array(list(reversed(itkimage.GetSpacing())))
+
+    return numpyImage, numpyOrigin, numpySpacing