Update nodes.py

2024-07-30 22:51:45 +03:00
parent 43e2074f45
commit c8fdfd6dab
1 changed files with 2 additions and 0 deletions
@@ -304,6 +304,8 @@ class Florence2Run:
                annotated_image_tensor = F.to_tensor(annotated_image_pil)
                out_tensor = annotated_image_tensor[:3, :, :].unsqueeze(0).permute(0, 2, 3, 1).cpu().float()
                out.append(out_tensor)
+               
+                out_data.append(bboxes)

                
                pbar.update(1)