Los datos en formato json de mmdetection se convierten en execl

Después de que terminé de entrenar con mmdetection, los resultados de nuestro entrenamiento están en el archivo json, como se muestra a continuación en 20230105_221053.log.json Sin embargo, cuando usamos mmdetection para dibujar, encontramos que las imágenes no se ven muy bien. Por lo tanto, necesitamos exportar estos datos. El código exportado se ve así:

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)

Hice este código usando chatgpt y luego lo modifiqué, y se siente muy fácil de usar.