Skip to content

GitLab

Unverified Commit f20ab37d authored by Xiaomeng Zhao's avatar Xiaomeng Zhao Committed by GitHub
Browse files

Merge pull request #1822 from icecraft/fix/caption_match 

fix: caption match
parents 98adcbb9 fb02be19
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
magic_pdf/model/magic_model.py
View file @ f20ab37d
...@@ -528,14 +528,13 @@ class MagicModel: ...@@ -528,14 +528,13 @@ class MagicModel:
pair_dis = bbox_distance(subjects[sub_idx]['bbox'], objects[obj_idx]['bbox']) pair_dis = bbox_distance(subjects[sub_idx]['bbox'], objects[obj_idx]['bbox'])
nearest_dis = float('inf') nearest_dis = float('inf')
for i in range(N): for i in range(N):
if i in seen_idx:continue if i in seen_idx or i == sub_idx:continue
nearest_dis = min(nearest_dis, bbox_distance(subjects[i]['bbox'], objects[obj_idx]['bbox'])) nearest_dis = min(nearest_dis, bbox_distance(subjects[i]['bbox'], objects[obj_idx]['bbox']))
if pair_dis >= 3*nearest_dis: if pair_dis >= 3*nearest_dis:
seen_idx.add(sub_idx) seen_idx.add(sub_idx)
continue continue
seen_idx.add(sub_idx) seen_idx.add(sub_idx)
seen_idx.add(obj_idx + OBJ_IDX_OFFSET) seen_idx.add(obj_idx + OBJ_IDX_OFFSET)
seen_sub_idx.add(sub_idx) seen_sub_idx.add(sub_idx)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment