论文画图python-yolov5折线图实验对比图

最新推荐文章于 2024-09-10 18:33:15 发布

原创

最新推荐文章于 2024-09-10 18:33:15 发布 · 2.4k 阅读

45 ·

CC 4.0 BY-SA版权

文章标签：

#python #matplotlib

import csv
import matplotlib.pyplot as plt
import pandas as pd
import numpy  as np
from urllib.parse import quote

plt.rcParams['font.sans-serif'] = ['SimHei']  # 用来正常显示中文标签
plt.rcParams['axes.unicode_minus'] = False    # 用来正常显示负号
df1 = pd.read_csv(r"D:\yolov5-master - newway\yolov5-master\runs\train\exp57\results.csv")   #读取文件1
df2 = pd.read_csv(r"D:\yolov5-master - newway\yolov5-master\runs\train\exp72\results.csv")   #读取文件2

epoch_1 = df1["               epoch"].values.tolist()                     #通过文件表头信息读取文件内容
mAP5_1  = df1["      train/cls_loss"].values.tolist()

epoch_2 = df2["               epoch"].values.tolist()                     #通过文件表头信息读取文件内容
mAP5_2  = df2["      train/cls_loss"].values.tolist()
# fig=plt.figure(dpi=128,figsize=(10,10))
plt.figure(dpi=200,figsize=(8,5))
plt.plot(epoch_1,mAP5_1,color='red',  label='yolov5s改进算法')       #设置曲线相关系数
plt.plot(epoch_2,mAP5_2,color='black',label='yolov5s原始算法')       #设置曲线相关系数

plt.xticks(fontsi