这里仅仅考虑person类别,如果考虑其他类别,则需要增加类别信息,稍作调整即可。
import json
import os
import imagesize
import copy
def txt_to_json(img_dir,annotation_dir,json_path,img_format='.jpg',annotation_format='.txt'):
# json 文件主要两项内容
json_dict = dict()
annotations = list()
images = list()
categories = list()
# 一个标签和一张图
one_annotation = dict()
one_image = dict()
annotation_bbox_id = 0
for file in os.listdir(annotation_dir):
if file.endswith(annotation_format):
# 读取图片信息:长宽,整合到one_image当中
one_image[