nodes/kijai-ComfyUI-Florence2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
kijai
2024-06-19 21:40:11 +03:00
parent c4bf85c143
commit e435feb08e
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+2 -2
View File
@@ -144,7 +144,7 @@ class Florence2Run:
results = processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
out_results.append(results)
if task == 'annotate' or task == 'dense_region_caption':
if task == 'annotate' or task == 'dense_region_caption' or task == 'caption_to_phrase_grounding':
parsed_answer = processor.post_process_generation(results, task="<OD>", image_size=(image_pil.width, image_pil.height))
fig, ax = plt.subplots(figsize=(image_pil.width / 100, image_pil.height / 100), dpi=100)
@@ -195,7 +195,7 @@ class Florence2Run:
plt.close(fig)
elif task == 'referring_expression_segmentation' or task == 'caption_to_phrase_grounding':
elif task == 'referring_expression_segmentation':
parsed_answer = processor.post_process_generation(results, task="<REFERRING_EXPRESSION_SEGMENTATION>", image_size=(image_pil.width, image_pil.height))
width, height = image_pil.size
# Create a new black image