调整json文件

全息数据

于 2022-10-16 16:33:30 发布

阅读量321

点赞数 1

分类专栏：深度学习图像算法图像分割文章标签： json 图像处理深度学习

本文链接：https://blog.csdn.net/qq_23022733/article/details/127349463

版权

深度学习同时被 3 个专栏收录

71 篇文章 10 订阅

订阅专栏

图像算法

60 篇文章 4 订阅

订阅专栏

图像分割

21 篇文章 3 订阅

订阅专栏

代码：

"""
针对行李箱原先的数据集，调整有无拉杆的标注
"""
import json

data_dir = {"labels": []}
json_path = r"C:\Users\9ling\Desktop\LuggageDataset_hasrod\test\has_luggage\baiyun_airport\baiyun_airport.json"
data = json.load(open(json_path))["labels"]
final_json = r"C:\Users\9ling\Desktop\LuggageDataset_hasrod\test\has_luggage\baiyun_airport.json"
for sanno in data:
    data_dir["labels"].append({"annotations": [], "filename": sanno["filename"]})
    for pes in sanno["annotations"]:
        if pes["class"] == "pedestrain":
            luggage_anno = pes["subclass"]
            data_dir["labels"][-1]["annotations"].append(pes)
            data_dir["labels"][-1]["annotations"].extend(pes["subclass"])

json_test_str = json.dumps(data_dir, indent=4)
with open(final_json, 'w') as json_file:
    json_file.write(json_test_str)