import argparse
import sys
import matplotlib.pyplot as plt
def main(argv):
parser = argparse.ArgumentParser()
parser.add_argument(
"-file",
help = "path to log file"
)
args = parser.parse_args()
f = open(args.file)
lines = [line.rstrip("\n") for line in f.readlines()]
numbers = {'1','2','3','4','5','6','7','8','9'}
iters = []
loss = []
fig,ax = plt.subplots()
prev_line = ""
for line in lines:
args = line.split(' ')
if args[0][-1:]==':' and args[0][0] in numbers :
iters.append(int(args[0][:-1]))
loss.append(float(args[2]))
ax.plot(iters,loss)
plt.xlabel('iters')
plt.ylabel('loss')
plt.grid()
ticks = range(0,250,10)
#ax.set_yticks(ticks)
plt.show()
if __name__ == "__main__":
main(sys.argv)
Darkent版yolo训练日志可视化脚本
最新推荐文章于 2024-08-27 12:28:12 发布