yolov5训练结果中使用plot画图的代码

以下是在Yolov5训练结果中使用plot画图的示例代码：

import matplotlib.pyplot as plt
import numpy as np

# 读取训练结果文件
with open('results.txt', 'r') as f:
    lines = f.readlines()

# 解析训练结果中的损失和精度
losses = []
accuracies = []
for line in lines:
    if 'val_loss' in line:
        loss = float(line.split('val_loss = ')[-1])
        losses.append(loss)
    elif 'val_acc' in line:
        accuracy = float(line.split('val_acc = ')[-1])
        accuracies.append(accuracy)

# 绘制损失和精度曲线
epochs = np.arange(1, len(losses)+1)
fig, ax = plt.subplots(2, 1, figsize=(10, 10))
ax[0].plot(epochs, losses, label='val_loss')
ax[0].set_title('Validation Loss')
ax[0].set_xlabel('Epoch')
ax[0].set_ylabel('Loss')
ax[0].legend()
ax[1].plot(epochs, accuracies, label='val_acc')
ax[1].set_title('Validation Accuracy')
ax[1].set_xlabel('Epoch')
ax[1].set_ylabel('Accuracy')
ax[1].legend()

plt.show()

在这个示例代码中，我们首先读取了训练结果文件，解析出了每个epoch的损失和精度。然后使用matplotlib库中的plot函数绘制了损失和精度曲线。最后使用show函数展示了绘制出的图像。