烂笔头3_Matplotlib 可视化模型表现基本数据
import pandas as pd
from matplotlib import pyplot as plt
import os
path = r"D:\results.csv"
df = pd.read_csv(path)
df.head()
![数据前5行](https://img-blog.csdnimg.cn/792fd95a38c54fc0b6f978b091b214b9.png)
plt.figure(figsize=(30,15))
plt.title('Model Performacne',fontsize=30)
plt.plot(df['epoch'],df['metrics/mAP_0.5'])
plt.plot(df['metrics/mAP_0.5:0.95'])
plt.scatter(df['epoch'],df['metrics/precision'],color='red',s=30)
plt.xlabel('Epoch',fontsize= 25)
plt.tick_params(labelsize=23)
plt.legend(['mAP_0.5','mAP_0.5:0.95','precision'],fontsize=20)
plt.show()
![结果图片](https://img-blog.csdnimg.cn/48ae50b5e93d483dbcfe6ccff4230cac.png)
-
4. 总结
上述提供的代码很简单,但是已经能都可视化基本的模型训练表现了,记录平时自己遇到并解决的问题,如有帮助欢迎一键三连!