spdraptor commited on
Commit
71e295f
·
verified ·
1 Parent(s): 905b8aa

Update modules/masking_module.py

Browse files
Files changed (1) hide show
  1. modules/masking_module.py +1 -1
modules/masking_module.py CHANGED
@@ -137,7 +137,7 @@ def masking_process(image,obj):
137
  # task_prompt = '<REGION_TO_SEGMENTATION>'
138
  # # task_prompt = '<OPEN_VOCABULARY_DETECTION>'
139
  # print(type(task_prompt),type(obj))
140
- image = image.convert("RGB")
141
 
142
  # results = florence2(image,task_prompt, text_input=obj)
143
  # output_image = copy.deepcopy(image)
 
137
  # task_prompt = '<REGION_TO_SEGMENTATION>'
138
  # # task_prompt = '<OPEN_VOCABULARY_DETECTION>'
139
  # print(type(task_prompt),type(obj))
140
+ image = Image.fromarray(image).convert("RGB")
141
 
142
  # results = florence2(image,task_prompt, text_input=obj)
143
  # output_image = copy.deepcopy(image)