From 6ea76277c9157cbe542fa20a69351d4a1afc4e70 Mon Sep 17 00:00:00 2001 From: Junfeng Wu <44546837+wjf5203@users.noreply.github.com> Date: Wed, 8 May 2024 23:01:01 +0800 Subject: [PATCH] Update merge_sa1b.py --- conversion/merge_sa1b.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/conversion/merge_sa1b.py b/conversion/merge_sa1b.py index 7219956..40f77f1 100644 --- a/conversion/merge_sa1b.py +++ b/conversion/merge_sa1b.py @@ -7,8 +7,8 @@ annbase = json.load(open('sa_000000_joint.json','rb')) for i in range(len(annbase['images'])): annbase['images'][i]['file_name'] = 'sa_000000/' + annbase['images'][i]['file_name'] -for i in range(1,50): - ann_i = json.load(open('sa_000{}_joint.json'.format(str(i).zfill(3)),'rb')) +for file_idx in range(1,50): + ann_i = json.load(open('sa_000{}_joint.json'.format(str(file_idx).zfill(3)),'rb')) for i in range(len(ann_i['images'])): ann_i['images'][i]['file_name'] = 'sa_000{}/'.format(str(i).zfill(3)) + ann_i['images'][i]['file_name'] annbase['images'] += ann_i['images']