许瑞 преди 1 година
родител
ревизия
d59981499d
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      magic_pdf/pre_proc/remove_bbox_overlap.py

+ 1 - 1
magic_pdf/pre_proc/remove_bbox_overlap.py

@@ -64,7 +64,7 @@ def _remove_overlap_between_bboxes(arr):
             else:
                 if v["score"] > res[i]["score"]:
                     keeps[i] = False
-                    res[i] = False
+                    res[i] = None
                 else:
                     keeps[idx] = False
                 drop_reasons.append(drop_reasons)