From be990f02507c3182701eb3d891436ce3bd4264fb Mon Sep 17 00:00:00 2001 From: Bao Anchang <74052592+adfwer233@users.noreply.github.com> Date: Wed, 19 Apr 2023 20:09:05 +0800 Subject: [PATCH 1/2] add rgba to rgb convert in edit_anything_model --- models/segment_models/edit_anything_model.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/models/segment_models/edit_anything_model.py b/models/segment_models/edit_anything_model.py index afeb3b3..5114448 100644 --- a/models/segment_models/edit_anything_model.py +++ b/models/segment_models/edit_anything_model.py @@ -58,5 +58,7 @@ def region_level_semantic_api(self, image, anns, topk=5): def semantic_class_w_mask(self, img_src, anns): image = Image.open(img_src) + if (image.mode == "RGBA"): + image = image.convert("RGB") image = resize_long_edge(image, 384) - return self.region_level_semantic_api(image, anns) \ No newline at end of file + return self.region_level_semantic_api(image, anns) From bdd178173a104a7e40545471bd882c0ed599dc7c Mon Sep 17 00:00:00 2001 From: Bao Anchang <74052592+adfwer233@users.noreply.github.com> Date: Wed, 19 Apr 2023 20:15:21 +0800 Subject: [PATCH 2/2] delete rgba if --- models/segment_models/edit_anything_model.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/models/segment_models/edit_anything_model.py b/models/segment_models/edit_anything_model.py index 5114448..3b995d2 100644 --- a/models/segment_models/edit_anything_model.py +++ b/models/segment_models/edit_anything_model.py @@ -58,7 +58,6 @@ def region_level_semantic_api(self, image, anns, topk=5): def semantic_class_w_mask(self, img_src, anns): image = Image.open(img_src) - if (image.mode == "RGBA"): - image = image.convert("RGB") + image = image.convert("RGB") image = resize_long_edge(image, 384) return self.region_level_semantic_api(image, anns)