caffe环境下，画faster rcnn的训练loss曲线图

最新推荐文章于 2025-01-04 20:56:21 发布

春暖and花开

最新推荐文章于 2025-01-04 20:56:21 发布

阅读量515

点赞数 1

分类专栏：深度学习

本文链接：https://blog.youkuaiyun.com/qq_34590704/article/details/95902652

版权

用faster rcnn对自己的数据集训练，得到训练模型的同时，训练日志也会自动保存在./experiments/logs文件夹下。要想知道整个训练过程中训练损失的变化，可通过画Loss曲线图观察，并以此调整学习率、迭代次数等参数。具体代码如下：

#!/usr/bin/env python

import os

import sys

import numpy as np

import matplotlib.pyplot as plt

import math

import re

import pylab

from pylab import figure, show, legend

from mpl_toolkits.axes_grid1 import host_subplot

fp = open('E:/fastrcnn/py-faster-rcnn/experiments/logs/faster_rcnn_end2end_VGG16_.txt.2019-07-14_11-25-04', 'r')  
# 改成自己的日志文件地址

train_iterations = []

train_loss = []

test_iterations = []

#test_accuracy = []

for ln in fp:

      if '] Iteration ' in ln and 'loss = ' in ln:

        arr = re.findall(r'ion \b\d+\b,',ln)

        train_iterations.append(int(arr[0].strip(',')[4:]))

        train_loss.append(float(ln.strip().split(' = ')[-1]))

fp.close()

host = host_subplot(111)