许瑞 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)