简介:该程序实现将第二组数据集中的每张图片的信息,分别写入以该图片名命名的json文件中,这样更方便进行图像标注和处理。同时,为了避免程序无法找到"data"文件夹而报错,程序中加入了创建文件夹的代码。
使用该程序可以很方便地将一些大型的数据集中的图像信息分离出来,方便后续对每张图片的处理和标注。同时,程序可读性较高,易于理解和修改。
import json
import os
# 读取json文件
with open("1.json", 'r') as f:
data = json.load(f)
# 遍历数据集中的每个图片
for key in data:
# 获取文件名
filename = data[key]['filename']
# 构造新的json格式数据
new_data = {
"filename": filename,
"size": data[key]['size'],
"regions": data[key]['regions'],
"file_attributes": {}
}
# 将新数据写入json文件
with open(os.path.join("data", filename.split(".")[0] + ".json"), 'w') as f:
json.dump(new_data, f)