mmdetection的json格式的数据转换成execl当中

我用mmdetection训练完成之后,我们训练的结果在json文件当中,如下所示在20230105_221053.log.json当中,但是,当我们使用mmdetection画图时发现画的不是很好看。因此,我们需要把这些数据导出来。导出的代码如下所示:

import json
import pandas as pd

# 打开txt文件并读取其中的JSON数据
with open('20230105_221053.log.json', 'r') as f:
    json_data = f.readlines()

# 创建一个空的列表来保存所有包含loss_cls的JSON数据
loss_cls_data = []

for data in json_data:
    # 将JSON数据转换为Python字典对象
    data_dict = json.loads(data)
    # 获取loss_cls的值
    # loss_cls = data_dict.get('loss_cls', None)
    # 如果loss_cls的值存在,则将其添加到列表中
    # if loss_cls:
    loss_cls_data.append(data_dict)

# 使用pandas模块创建DataFrame并将数据保存到Excel文件中
df = pd.DataFrame(loss_cls_data)
df.to_excel('loss_cls_data.xlsx', index=False)

我这个代码使用chatgpt搞得,然后修改了一下,感觉挺好用的。

猜你喜欢

转载自blog.csdn.net/qq_44666320/article/details/130103690