将json文件(coco的标签)写入csv文件

import json
import csv
import pandas as pd

# header = ['img_id','label','score','x1','x2','y1','y2']
# 读取json文件
with open('path.json', 'r') as f:
    data = json.load(f)

image_id = []
cate = []
score = []
x1m = []
x2m =[]
y1m = []
y2m = []

# with open('rcnn.csv','w',encoding='UTF8') as f:
#     writer = csv.writer(f)
#     writer.writerow(header)

# 循环控制写入单元格
for list_item in data:
    for key, value in list_item.items():
        if key == "image_id":
            image_id.append(value)
        elif key == "category_id":
            cate.append(value)
        elif key == "score":
            score.append( value)
        elif key == "bbox":
            x1 = value[0]
            x2 = value[0]+value[2]
            y1 = value[1]
            y2 = value[1]+value[3]
            x1m.append(x1)
            x2m.append(x2)
            y1m.append(y1)
            y2m.append(y2)

dateframe = pd.DataFrame({"img_id":image_id,'label':cate,'score':score,'x1':x1m,'x2':x2m,'y1':y1m,'y2':y2m})
dateframe.to_csv('path.csv',sep=',',index=False)
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值