import csv
from matplotlib import pyplot as plt
import numpy as np
#将文件名存储在filename中
filename='/home/use4test/yl/yolov5/runs/train/exp48/results.csv'
#将文件名存储在filename中
with open(filename) as f:
#创建阅读器(调用csv.reader()将前面存储的文件对象最为实参传给它)
reader=csv.reader(f)
#调用了next()一次,所以这边只调用了文件的第一行,并将头文件存储在header_row中
header_row=next(reader)
dates,highs=[],[]
for row in reader:
date=int(row[0])
dates.append(date)
high=float(row[6])
highs.append(high)
#根据数据绘制图形
fig=plt.figure(dpi=128,figsize=(10,10))
plt.plot(dates,highs,c='red')
#设置图形格式
plt.title("",fontsize=24)
plt.xlabel('epoch',fontsize=24)
fig.autofmt_xdate()
new_ticks = np.linspace(0,1,11)
plt.xlim((0,300))
plt.ylim((0,1))
plt.yticks(new_ticks)
plt.ylabel('',fontsize=16)
plt.tick_params(axis='both',which='major',labelsize=16)
plt.show()
YOLOv5学习实战四:基于Matplotlib用于论文结果分析的result.csv文件可视化
最新推荐文章于 2024-06-13 14:17:51 发布