OD-数据预处理-对fsod-test和fsod-train数据集进行统计

import json

with open('fsod_test.json') as fd:
    file=json.load(fd)

# print(file)
print(type(file))
keys=file.keys()
for k in keys:
    print(k)
    print(type(file[k]))
supcat=[]
id=[]
name=[]
slow=[]
cats=file['categories']
for item in cats:
    supcat.append(item['supercategory'])
    name.append(item['name'])
    id.append(item['id'])

print(set(supcat))
print('--'*50)
print(set(id))
print('--'*50)
print(set(name))
with open('count_log_train.txt','a') as f:
    tmpp=set(supcat)
    tmp=[i for i in tmpp]
    f.write('supcat   '+str(len(tmp))+'\n')
    for i,j in enumerate(tmp):
        f.write(tmp[i]+'  ')
        if(i%10==0):
            f.write('\n')
    tmpp=set(id)
    tmp = [i for i in tmpp]
    f.write('id  '+str(len(tmp))+'\n')
    for i, j in enumerate(tmp):
        f.write(str(tmp[i]) + '  ')
        if (i % 10 == 0):
            f.write('\n')

    tmpp=set(name)
    tmp = [i for i in tmpp]
    f.write('name   '+ str(len(tmp))+'\n')
    for i, j in enumerate(tmp):
        f.write(tmp[i] + '  ')
        if (i % 10 == 0):
            f.write('\n')

查看训练集和测试集的类别数量和详细信息以及文件的json文件的组织结构