import json
import glob

def build_basic_dict():
? ?basic_dict = {}

? ?info = {}
? ?description = 'small_person_keypoints'
? ?author = 'lishaohua'
? ?info.update({'description':description,'author':author})
? ?basic_dict.update({'info':info})

? ?categories={}
? ?categories.update({'supercategory':'person','id':1,'name':'person'})
? ?keypoints = ["nose",
? ? ? ? ? ? ? ? "left_eye","right_eye",
? ? ? ? ? ? ? ? "left_ear","right_ear",
? ? ? ? ? ? ? ? "left_shoulder","right_shoulder",
? ? ? ? ? ? ? ? "left_elbow","right_elbow",
? ? ? ? ? ? ? ? "left_wrist","right_wrist",
? ? ? ? ? ? ? ? "left_hip","right_hip",
? ? ? ? ? ? ? ? "left_knee","right_knee",
? ? ? ? ? ? ? ? "left_ankle","right_ankle"]
? ?skeleton = [[16,14],
? ? ? ? ? ? ? ?[14,12],
? ? ? ? ? ? ? ?[17,15],
? ? ? ? ? ? ? ?[15,13],
? ? ? ? ? ? ? ?[12,13],
? ? ? ? ? ? ? ?[6,12],
? ? ? ? ? ? ? ?[7,13],
? ? ? ? ? ? ? ?[6,7],
? ? ? ? ? ? ? ?[6,8],
? ? ? ? ? ? ? ?[7,9],
? ? ? ? ? ? ? ?[8,10],
? ? ? ? ? ? ? ?[9,11],
? ? ? ? ? ? ? ?[2,3],
? ? ? ? ? ? ? ?[1,2],
? ? ? ? ? ? ? ?[1,3],
? ? ? ? ? ? ? ?[2,4],
? ? ? ? ? ? ? ?[3,5],
? ? ? ? ? ? ? ?[4,6],
? ? ? ? ? ? ? ?[5,7]]
? ?categories.update({'keypoints':keypoints,'skeleton':skeleton})
? ?basic_dict.update({'categories':categories})
? ?return basic_dict

def build_images(img_lists,json_lists):
? ?images=[]
? ?for name in img_lists:
? ? ? ?file_name = name.split('\\')[7]
? ? ? ?id = int(file_name.split('.')[0])
? ? ? ?for j in json_lists:
? ? ? ? ? ?with open(j, 'r') as f:
? ? ? ? ? ? ? ?f = json.load(f)
? ? ? ? ? ? ? ?if file_name == f['imagePath']:
? ? ? ? ? ? ? ? ? ?height = f['imageHeight']
? ? ? ? ? ? ? ? ? ?width = f['imageWidth']
? ? ? ? ? ? ? ? ? ?image={}
? ? ? ? ? ? ? ? ? ?image.update({'file_name':file_name,'height':height,'width':width,'id':id})
? ? ? ? ? ? ? ? ? ?images.append(image)
? ?return images

def build_annotations(img_lists,json_lists):
? ?annotations_dict = {}
? ?annotations_dicts =[]
? ?ann_id = 0
? ?for i in json_lists:
? ? ? ?with open(i, 'r') as f:
? ? ? ? ? ?shape = json.load(f)['shapes']
? ? ? ? ? ?num_person = int(len(shape) / (17 ? 1))
? ? ? ? ? ?keypoints= []
? ? ? ? ? ?bbox = []
? ? ? ? ? ?num_joint=0
? ? ? ? ? ?for per_joint in shape:
? ? ? ? ? ? ? ?if per_joint['group_id'] == 0:
? ? ? ? ? ? ? ? ? ?x,y,z = 0,0,0
? ? ? ? ? ? ? ? ? ?keypoints.extend([x,y,z])
? ? ? ? ? ? ? ? ? ?num_joint ?=1
? ? ? ? ? ? ? ?if 'bbox' in per_joint['label']:
? ? ? ? ? ? ? ? ? ?xmin,ymin = enumerate(per_joint['points'][0])
? ? ? ? ? ? ? ? ? ?xmax,ymax = enumerate(per_joint['points'][1])
? ? ? ? ? ? ? ? ? ?bbox.extend([xmin[1],ymin[1],xmax[1]-xmin[1],ymax[1]-ymin[1]])
? ? ? ? ? ? ? ?if per_joint['group_id'] == 1:
? ? ? ? ? ? ? ? ? ?x,y = enumerate(per_joint['points'][0])
? ? ? ? ? ? ? ? ? ?keypoints.extend([x[1],y[1],per_joint['group_id']])
? ? ? ? ? ? ? ?if per_joint['group_id'] == 2 and 'bbox' not in per_joint['label']:
? ? ? ? ? ? ? ? ? ?x, y = enumerate(per_joint['points'][0])
? ? ? ? ? ? ? ? ? ?keypoints.extend([x[1], y[1], per_joint['group_id']])
? ? ? ? ? ?num_keypoints = 17 - num_joint
? ? ? ? ? ?image_id = int(i.split('\\')[-1].split('.')[0])
? ? ? ? ? ?category_id = 1
? ? ? ? ? ?for j in range(num_person):
? ? ? ? ? ? ? ?ann_id ?= 1
? ? ? ? ? ? ? ?annotations_dict.update({'num_keypoints':num_keypoints,
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? 'area':int(bbox[2] * bbox[3]),
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? 'iscrowd': int(0),
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? 'keypoints':keypoints[16*3*j:16*3*(j 1) 3],
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? 'image_id':image_id,
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? 'bbox':bbox[0:4],
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? 'category_id':category_id,
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? 'id':int(ann_id-1)})
? ? ? ? ? ? ? ?a = annotations_dict.copy()
? ? ? ? ? ? ? ?annotations_dicts.append(a)
? ?return annotations_dicts

file_path = 'C:\\Users\\zstu\\Desktop\\smallperson\\images\\val2017\\' ?#需要修改
json_lists = glob.glob(file_path + '\\*.json')
img_lists = glob.glob(file_path + '\\*.jpg')

for json_name in json_lists:
? ?with open(json_name,'r') as f:
? ? ? ?f = json.load(f)

? ?basic_dicts = build_basic_dict()
? ?images_dicts = build_images(img_lists,json_lists)
? ?annotations_dicts = build_annotations(img_lists,json_lists)
? ?tinyperson = {}
? ?tinyperson.update({'info':basic_dicts['info'],'images':images_dicts,'annotations':annotations_dicts,'categories':[basic_dicts['categories']]})
? ?json.dump(tinyperson,open('C:\\Users\\zstu\\Desktop\\smallperson\\annotations\\tinyperson.json','w'),indent=4) #需要修改