Fix infer point bug

Signed-off-by: heyufan1995 <[email protected]>
Project-MONAI · Jul 17, 2024 · 1090efb · 1090efb
1 parent d71a2d1
commit 1090efb
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/scripts/infer.py b/scripts/infer.py
@@ -173,6 +173,9 @@ def infer(
             batch_data = list_data_collate([batch_data])
             self.batch_data = batch_data
         if point is not None:
+            if type(point) is list:
+                point = np.array(point)[np.newaxis, ...]
+                point_label = np.array(point_label)[np.newaxis, ...]
             point = self.transform_points(
                 point,
                 np.linalg.inv(batch_data["image"].affine[0])