# time:2022/11/29 10:09 import matplotlib.pyplot as plt import numpy as np #----------- 打开txt文件 ---------- file1 = open('D:/Net/epoch_miou.txt') file2 = open('D:/Net/epoch_val_miou.txt') # ----------- 逐行读取文件内的数据 ------------ data1 = file1.readlines() data2 = file2.readlines() #------ x轴数据有序生成150个(根据自己的横坐标范围自己修改范围) ---- x1 = np.arange(0,150) #---------- 新建一个空的列表,用于存储上一步逐行读取的data ------------ y1 = [] y2 = [] #---------- 用循环的方式添加进列表 ----------- for num1 in data1: y1.append(float(num1.split(',')[0]))#TXT文本的第一列################################ for num2 in data2: y2.append(float(num2.split(',')[0]))#TXT文本的第一列################################ #--------------- 输出图 ---------------------- #--------- 可以理解为在图上加载x和y的数据 label为关于x和y曲线的标签------------ plt.plot(x1, y1, color='red',label='train mIoU') plt.plot(x1, y2, color='grey',label='val mIoU') # --------- x轴的小标题 ------------- plt.xlabel('Epoch') # --------- y轴的小标题 ------------- plt.ylabel('mIoU') plt.legend() plt.show()