فهرست منبع

Update mineru/backend/pipeline/pipeline_magic_model.py

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Xiaomeng Zhao 3 ماه پیش
والد
کامیت
300b691713
1فایلهای تغییر یافته به همراه1 افزوده شده و 1 حذف شده
  1. 1 1
      mineru/backend/pipeline/pipeline_magic_model.py

+ 1 - 1
mineru/backend/pipeline/pipeline_magic_model.py

@@ -276,7 +276,7 @@ class MagicModel:
 
             fst_idx, fst_kind, left_x, top_y = candidates[0]
             fst_bbox = subjects[fst_idx]['bbox'] if fst_kind == SUB_BIT_KIND else objects[fst_idx - OBJ_IDX_OFFSET]['bbox']
-            candidates.sort(key=lambda x: bbox_distance(fst_bbox,subjects[x[0]]['bbox']) if x[1] == SUB_BIT_KIND else bbox_distance(fst_bbox, objects[x[0] - OBJ_IDX_OFFSET]['bbox']))
+            candidates.sort(key=lambda x: bbox_distance(fst_bbox, subjects[x[0]]['bbox']) if x[1] == SUB_BIT_KIND else bbox_distance(fst_bbox, objects[x[0] - OBJ_IDX_OFFSET]['bbox']))
             nxt = None
 
             for i in range(1, len(candidates)):