xcs

utilitis.py

@@ -73,6 +73,15 @@ def unnormalize_box(bbox, width, height):
          width * (bbox[2] / 1000),
          height * (bbox[3] / 1000),
      ]
+
+def drop_null_bbox(dictionary):
+    keys_to_drop = []
+    for key, (_, _, bbox_values) in dictionary.items():
+        if all(value == 0.0 for value in bbox_values):
+            keys_to_drop.append(key)
+    for key in keys_to_drop:
+        del dictionary[key]
+
 def get_word(bboxes,image):
     x_min, y_min, x_max, y_max = bboxes
     roi = image.crop((x_min, y_min, x_max, y_max)) # Region of intrest
@@ -96,7 +105,7 @@ def get_Finale_results(offset_mapping,id2label,image,prediction_scores,predictio
     for index, prediction, score, bbox in Final_prediction:
         if prediction not in Final_results or score > Final_results[prediction][1]:
             Final_results[prediction] = (index, score,bbox)
-    #print(Final_results)
+    drop_null_bbox(Final_results)
     
     for final in Final_results:
         Kalma = get_word(Final_results[final][2],image)