riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

[SAM] fix sam slow test (#23376)

Browse Source 
* fix sam slow test

* oops

* fix error message
This commit is contained in:
Younes Belkada 2023-05-17 14:27:43 +02:00 committed by GitHub
parent 22a0769933
commit 3d3c7d4213
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/models/sam/processing_sam.py
View File

@ -208,7 +208,7 @@ class SamProcessor(ProcessorMixin):
input_points = input_points.numpy().tolist()
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]
else:
input_points = None
@ -232,7 +232,7 @@ class SamProcessor(ProcessorMixin):
or not isinstance(input_boxes[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]
else:
input_boxes = None

4
tests/models/sam/test_modeling_sam.py
View File

@ -481,7 +481,7 @@ class SamModelIntegrationTest(unittest.TestCase):
model.eval()
raw_image = prepare_image()
input_boxes = [[650, 900, 1000, 1250]]
input_boxes = [[[650, 900, 1000, 1250]]]
input_points = [[[820, 1080]]]
inputs = processor(
@ -541,7 +541,7 @@ class SamModelIntegrationTest(unittest.TestCase):
model.eval()
raw_image = prepare_image()
input_boxes = [[620, 900, 1000, 1255]]
input_boxes = [[[620, 900, 1000, 1255]]]
input_points = [[[820, 1080]]]
labels = [[0]]