Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "6c26faa159b79a42d7fa46cb66e2d21523351987"
Unverified Commit e6122c3f authored by amyeroberts's avatar amyeroberts Committed by GitHub
Browse files

Fix image segmentation tool bug (#23897) 

* Image segmentation tool bug

* Remove resizing in the tests
parent 6cd34d45
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 5 deletions
src/transformers/tools/image_segmentation.py
View file @ e6122c3f
...@@ -44,7 +44,6 @@ class ImageSegmentationTool(PipelineTool): ...@@ -44,7 +44,6 @@ class ImageSegmentationTool(PipelineTool):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
def encode(self, image: "Image", label: str): def encode(self, image: "Image", label: str):
self.pre_processor.image_processor.size = {"width": image.size[0], "height": image.size[1]}
return self.pre_processor(text=[label], images=[image], padding=True, return_tensors="pt") return self.pre_processor(text=[label], images=[image], padding=True, return_tensors="pt")
def forward(self, inputs): def forward(self, inputs):
......
tests/tools/test_image_segmentation.py
View file @ e6122c3f
...@@ -33,21 +33,21 @@ class ImageSegmentationToolTester(unittest.TestCase, ToolTesterMixin): ...@@ -33,21 +33,21 @@ class ImageSegmentationToolTester(unittest.TestCase, ToolTesterMixin):
self.remote_tool = load_tool("image-segmentation", remote=True) self.remote_tool = load_tool("image-segmentation", remote=True)
def test_exact_match_arg(self): def test_exact_match_arg(self):
image = Image.open(Path(get_tests_dir("fixtures/tests_samples/COCO")) / "000000039769.png").resize((512, 512)) image = Image.open(Path(get_tests_dir("fixtures/tests_samples/COCO")) / "000000039769.png")
result = self.tool(image, "cat") result = self.tool(image, "cat")
self.assertTrue(isinstance(result, Image.Image)) self.assertTrue(isinstance(result, Image.Image))
def test_exact_match_arg_remote(self): def test_exact_match_arg_remote(self):
image = Image.open(Path(get_tests_dir("fixtures/tests_samples/COCO")) / "000000039769.png").resize((512, 512)) image = Image.open(Path(get_tests_dir("fixtures/tests_samples/COCO")) / "000000039769.png")
result = self.remote_tool(image, "cat") result = self.remote_tool(image, "cat")
self.assertTrue(isinstance(result, Image.Image)) self.assertTrue(isinstance(result, Image.Image))
def test_exact_match_kwarg(self): def test_exact_match_kwarg(self):
image = Image.open(Path(get_tests_dir("fixtures/tests_samples/COCO")) / "000000039769.png").resize((512, 512)) image = Image.open(Path(get_tests_dir("fixtures/tests_samples/COCO")) / "000000039769.png")
result = self.tool(image=image, label="cat") result = self.tool(image=image, label="cat")
self.assertTrue(isinstance(result, Image.Image)) self.assertTrue(isinstance(result, Image.Image))
def test_exact_match_kwarg_remote(self): def test_exact_match_kwarg_remote(self):
image = Image.open(Path(get_tests_dir("fixtures/tests_samples/COCO")) / "000000039769.png").resize((512, 512)) image = Image.open(Path(get_tests_dir("fixtures/tests_samples/COCO")) / "000000039769.png")
result = self.remote_tool(image=image, label="cat") result = self.remote_tool(image=image, label="cat")
self.assertTrue(isinstance(result, Image.Image)) self.assertTrue(isinstance(result, Image.Image))
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