diff --git a/models/segment_models/edit_anything_model.py b/models/segment_models/edit_anything_model.py index afeb3b3..3b995d2 100644 --- a/models/segment_models/edit_anything_model.py +++ b/models/segment_models/edit_anything_model.py @@ -58,5 +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) + 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)