Skip to content

Commit 1090efb

Browse files
committed
Fix infer point bug
Signed-off-by: heyufan1995 <heyufan1995@gmail.com>
1 parent d71a2d1 commit 1090efb

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

scripts/infer.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -173,6 +173,9 @@ def infer(
173173
batch_data = list_data_collate([batch_data])
174174
self.batch_data = batch_data
175175
if point is not None:
176+
if type(point) is list:
177+
point = np.array(point)[np.newaxis, ...]
178+
point_label = np.array(point_label)[np.newaxis, ...]
176179
point = self.transform_points(
177180
point,
178181
np.linalg.inv(batch_data["image"].affine[0])

0 commit comments

Comments
 (0)