Update files/functions.py

files/functions.py

@@ -646,14 +646,13 @@ def predictions_line_level(dataset, outputs, images_ids_list, chunk_ids, input_i
             bbox_prev = [-100, -100, -100, -100]
             for probs, input_id, bbox in zip(ten_probs_list, ten_input_ids_list, ten_bboxes_list):
                 bbox = denormalize_box(bbox, width, height)
-                if bbox != bbox_prev and bbox != cls_box:
+                if bbox != bbox_prev and bbox != cls_box and bbox != sep_box and bbox[0] != bbox[2] and bbox[1] != bbox[3]:
                     bboxes_list.append(bbox)
                     input_ids_dict[str(bbox)] = [input_id]
                     probs_dict[str(bbox)] = [probs]
-                else:
-                    if bbox != cls_box:
-                        input_ids_dict[str(bbox)].append(input_id)
-                        probs_dict[str(bbox)].append(probs)
+                elif bbox != cls_box and bbox != sep_box and bbox[0] != bbox[2] and bbox[1] != bbox[3]:
+                    input_ids_dict[str(bbox)].append(input_id)
+                    probs_dict[str(bbox)].append(probs)
                 bbox_prev = bbox
 
             probs_bbox = dict()