许瑞 1 жил өмнө
parent
commit
d59981499d

+ 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)