added image detection code to display predicted bboxes

tool_utils/yolo_world.py

@@ -80,5 +80,5 @@ class YoloWorld:
                          )
         
         cv2.imwrite('final_mask.png', cv2.cvtColor(detected_image,cv2.COLOR_BGR2RGB))
-        return "Predicted image mask : final_mask.jpg . Details :{}".format(processed_predictions[0])
+        return "Predicted image mask : final_mask.jpg . Details :{}".format(processed_predictions[0]['boxes'])
                 

utils.py

@@ -49,5 +49,5 @@ def draw_bboxes(rgb_frame,boxes,labels,color=None,line_thickness=3):
         # label = label2id[int(label.numpy())]
         t_size = cv2.getTextSize(str(label), 0, fontScale=tl / 3, thickness=tf)[0]
         c2 = c1[0] + t_size[0], c1[1] - t_size[1] - 3
-        cv2.putText(rgb_frame_copy, str(label), (c1[0], c1[1] - 2), 0, tl / 3, [225, 255, 255], thickness=tf, lineType=cv2.LINE_AA)
+        cv2.putText(rgb_frame_copy, str(label), (c1[0], c1[1] - 2), 0, tl / 3, [225, 0, 255], thickness=tf, lineType=cv2.LINE_AA)
     return rgb_frame_copy