Merge pull request #1180 from SCIInstitute/fix_1179

Fix for #1179
SCIInstitute · Mar 30, 2021 · e839d2d · e839d2d
2 parents 4dd0ecc + 7fdae4e
commit e839d2d
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/Examples/Python/GroomUtils.py b/Examples/Python/GroomUtils.py
@@ -121,7 +121,9 @@ def FindReferenceImage(inDataList):
     """
     x = y = z = 0
     for i in range(len(inDataList)):
-        dim = itk.GetArrayFromImage(itk.imread(inDataList[i])).shape
+        img = Image(inDataList[i])
+        tmp = img.toArray()
+        dim = tmp.shape
         if dim[0] > x:
             x = dim[0]
         if dim[1] > y:
@@ -131,15 +133,17 @@ def FindReferenceImage(inDataList):
 
     COM = np.zeros((x, y, z))
     for i in range(len(inDataList)):
-        tmp = itk.GetArrayFromImage(itk.imread(inDataList[i]))
+        img = Image(inDataList[i])
+        tmp = img.toArray()
         COM += np.pad(tmp, (((x - tmp.shape[0]) // 2, (x - tmp.shape[0]) - (x - tmp.shape[0]) // 2),
                             ((y - tmp.shape[1]) // 2, (y - tmp.shape[1]) - (y - tmp.shape[1]) // 2),
                             ((z - tmp.shape[2]) // 2, (z - tmp.shape[2]) - (z - tmp.shape[2]) // 2)))
     COM /= len(inDataList)
     dist = np.inf
     idx = 0
     for i in range(len(inDataList)):
-        tmp = itk.GetArrayFromImage(itk.imread(inDataList[i]))
+        img = Image(inDataList[i])
+        tmp = img.toArray()
         tmp_dist = np.linalg.norm(
             COM - np.pad(tmp, (((x - tmp.shape[0]) // 2, (x - tmp.shape[0]) - (x - tmp.shape[0]) // 2),
                                ((y - tmp.shape[1]) // 2, (y - tmp.shape[1]) - (y - tmp.shape[1]) // 2),