Unverified Commit 3d3c7d42 authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

[`SAM`] fix sam slow test (#23376)

* fix sam slow test

* oops

* fix error message
parent 22a07699
...@@ -208,7 +208,7 @@ class SamProcessor(ProcessorMixin): ...@@ -208,7 +208,7 @@ class SamProcessor(ProcessorMixin):
input_points = input_points.numpy().tolist() input_points = input_points.numpy().tolist()
if not isinstance(input_points, list) or not isinstance(input_points[0], list): if not isinstance(input_points, list) or not isinstance(input_points[0], list):
raise ValueError("Input points must be a list of list of floating integers.") raise ValueError("Input points must be a list of list of floating points.")
input_points = [np.array(input_point) for input_point in input_points] input_points = [np.array(input_point) for input_point in input_points]
else: else:
input_points = None input_points = None
...@@ -232,7 +232,7 @@ class SamProcessor(ProcessorMixin): ...@@ -232,7 +232,7 @@ class SamProcessor(ProcessorMixin):
or not isinstance(input_boxes[0], list) or not isinstance(input_boxes[0], list)
or not isinstance(input_boxes[0][0], list) or not isinstance(input_boxes[0][0], list)
): ):
raise ValueError("Input boxes must be a list of list of list of floating integers.") raise ValueError("Input boxes must be a list of list of list of floating points.")
input_boxes = [np.array(box).astype(np.float32) for box in input_boxes] input_boxes = [np.array(box).astype(np.float32) for box in input_boxes]
else: else:
input_boxes = None input_boxes = None
......
...@@ -481,7 +481,7 @@ class SamModelIntegrationTest(unittest.TestCase): ...@@ -481,7 +481,7 @@ class SamModelIntegrationTest(unittest.TestCase):
model.eval() model.eval()
raw_image = prepare_image() raw_image = prepare_image()
input_boxes = [[650, 900, 1000, 1250]] input_boxes = [[[650, 900, 1000, 1250]]]
input_points = [[[820, 1080]]] input_points = [[[820, 1080]]]
inputs = processor( inputs = processor(
...@@ -541,7 +541,7 @@ class SamModelIntegrationTest(unittest.TestCase): ...@@ -541,7 +541,7 @@ class SamModelIntegrationTest(unittest.TestCase):
model.eval() model.eval()
raw_image = prepare_image() raw_image = prepare_image()
input_boxes = [[620, 900, 1000, 1255]] input_boxes = [[[620, 900, 1000, 1255]]]
input_points = [[[820, 1080]]] input_points = [[[820, 1080]]]
labels = [[0]] labels = [[0]]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment