MMSelfSup根据训练产生的json文件画出Loss曲线

1、MMSelfSup官方提供的analysis_logs运行会报错，于是使用以下程序来画loss曲线。
2、下图是训练过程中产生的json文件。可以看到一个epoch会产生多个loss值（epoch是按iter来划分的）。于是取一个epoch中，多个loss值的平均值作为这个epoch的平均loss值。
3、程序先根据epoch划分loss值，求出每个epoch对应的平均loss值。最后画出epoch与平均loss值的折线图。

import matplotlib.pyplot as plt
import json
from collections import defaultdict

# 读取JSON文件，按epoch分组并计算每个epoch的平均loss
def process_json_file(file_path):
    epoch_losses = defaultdict(list)

    with open(file_path, 'r') as f:
        for line in f:
            data = json.loads(line)
            epoch = data.get("epoch")
            loss = data.get("loss")

            if epoch is not None and loss is not None:
                epoch_losses[epoch].append(loss)

    # 计算每个epoch的平均loss
    epochs = sorted(epoch_losses.keys())
    average_losses = [sum(epoch_losses[epoch]) / len(epoch_losses[epoch]) for epoch in epochs]

    return epochs, average_losses

# JSON文件路径，根据你的实际路径进行修改
json_file_path = "/home/znck/PycharmProjects/mmselfsup-main/mmselfsup-main/checkpoints/20230920_231524/vis_data/scalars.json"

# 处理JSON文件并获取数据
epochs, average_losses = process_json_file(json_file_path)

# 绘制曲线图
plt.figure(figsize=(8, 6))
plt.plot(epochs, average_losses, linestyle='-')
plt.xlabel('Epoch')
plt.ylabel('Average Loss')
plt.title('Average Loss Over Epochs')
plt.grid(True)
plt.show()