Fix location retrieval

DIDSR · Apr 19, 2022 · 5a96db2 · 5a96db2
1 parent f901ce0
commit 5a96db2
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/Victre/Pipeline.py b/Victre/Pipeline.py
@@ -1383,13 +1383,13 @@ def insert_lesions(self, lesion_type=None, n=-1, lesion_file=None, lesion_size=N
                                               ]))
 
                 loc = {"dm": self.get_coordinates_dm([
-                    cand[1] + lesion.shape[1] / 2,
-                    cand[2] + lesion.shape[2] / 2,
-                    cand[0] + lesion.shape[0] / 2]),
+                    cand[1],
+                    cand[2],
+                    cand[0]]),
                     "dbt": self.get_coordinates_dbt([
-                        cand[1] + lesion.shape[1] / 2,
-                        cand[2] + lesion.shape[2] / 2,
-                        cand[0] + lesion.shape[0] / 2])}
+                        cand[1],
+                        cand[2],
+                        cand[0]])}
 
                 self.lesion_locations["dm"].append(
                     list(np.round([loc["dm"][0], loc["dm"][1], cand_type]).astype(int)))