YOLOv5学习实战四：基于Matplotlib用于论文结果分析的result.csv文件可视化

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()