Commit f731fcab authored by icecraft's avatar icecraft
Browse files

fix: update figure caption match algorithm

parent 0793da41
...@@ -518,13 +518,13 @@ class MagicModel: ...@@ -518,13 +518,13 @@ class MagicModel:
break break
if nxt is None: if nxt is None:
break break
if fst_kind == SUB_BIT_KIND: if fst_kind == SUB_BIT_KIND:
sub_idx, obj_idx = fst_idx, nxt[0] - OBJ_IDX_OFFSET sub_idx, obj_idx = fst_idx, nxt[0] - OBJ_IDX_OFFSET
else: else:
sub_idx, obj_idx = nxt[0], fst_idx - OBJ_IDX_OFFSET sub_idx, obj_idx = nxt[0], fst_idx - OBJ_IDX_OFFSET
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):
...@@ -533,13 +533,13 @@ class MagicModel: ...@@ -533,13 +533,13 @@ class MagicModel:
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)
ret.append( ret.append(
{ {
'sub_bbox': { 'sub_bbox': {
......
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