如果我要将打印结果用折线图表示出来请给出代码

如果你想将模型训练过程中IoU的变化趋势以折线图的形式展示出来，可以使用Python的matplotlib库。下面是一个简单的例子，假设你已经保存了每个epoch的平均IoU：

import matplotlib.pyplot as plt

# 假设你有一个字典存储每个epoch的IoU结果
iou_history = {
    'Epoch 1': 0.65,
    'Epoch 2': 0.72,
    'Epoch 3': 0.78,
    # 更多epochs...
}

# 获取epochs和对应的IoU值
epochs = list(iou_history.keys())
ious = [iou_history[epoch] for epoch in epochs]

# 创建折线图
plt.figure(figsize=(10, 6))
plt.plot(epochs, ious, label='Average IoU')
plt.xlabel('Epochs')
plt.ylabel('IoU Score')
plt.title('Training IoU Over Time')
plt.grid(True)
plt.legend()

# 显示图形或保存至文件
plt.show()  # 如果在Jupyter Notebook中，会自动显示
# 或者 savefig("training_iou.png")

这个脚本会创建一个图表，横轴表示训练轮数，纵轴表示IoU分数。你可以根据实际情况调整代码，例如从日志文件读取历史数据，或者实时更新折线图。