Caffe学习系列(19): 绘制loss和accuracy曲线
来源:互联网 发布:window10无法安装软件 编辑:程序博客网 时间:2024/06/04 20:14
如同前几篇的可视化,这里采用的也是jupyter notebook来进行曲线绘制。
In [1]:
#加载必要的库import numpy as npimport matplotlib.pyplot as plt%matplotlib inlineimport sys,os,caffe#设置当前目录caffe_root = '/home/bnu/caffe/' sys.path.insert(0, caffe_root + 'python')os.chdir(caffe_root)
设置求解器,和c++/caffe一样,需要一个solver配置文件。
In [2]:
# set the solver prototxtcaffe.set_device(0)caffe.set_mode_gpu()solver = caffe.SGDSolver('examples/cifar10/cifar10_quick_solver.prototxt')
如果不需要绘制曲线,只需要训练出一个caffemodel, 直接调用solver.solve()就可以了。如果要绘制曲线,就需要把迭代过程中的值
保存下来,因此不能直接调用solver.solve(), 需要迭代。在迭代过程中,每迭代200次测试一次
In [5]:
%%timeniter =4000test_interval = 200train_loss = np.zeros(niter)test_acc = np.zeros(int(np.ceil(niter / test_interval)))# the main solver loopfor it in range(niter): solver.step(1) # SGD by Caffe # store the train loss train_loss[it] = solver.net.blobs['loss'].data solver.test_nets[0].forward(start='conv1') if it % test_interval == 0: acc=solver.test_nets[0].blobs['accuracy'].data print 'Iteration', it, 'testing...','accuracy:',acc test_acc[it // test_interval] = acc
绘制train过程中的loss曲线,和测试过程中的accuracy曲线
print test_acc_, ax1 = plt.subplots()ax2 = ax1.twinx()ax1.plot(np.arange(niter), train_loss)ax2.plot(test_interval * np.arange(len(test_acc)), test_acc, 'r')ax1.set_xlabel('iteration')ax1.set_ylabel('train loss')ax2.set_ylabel('test accuracy')
[ 0.1 0.41999999 0.47999999 0.54000002 0.62 0.63 0.64999998 0.66000003 0.66000003 0.67000002 0.70999998 0.69999999 0.75 0.74000001 0.76999998 0.75 0.69999999 0.74000001 0.72000003 0.76999998]
Out[6]:
在[2]时,出现The kernel appears to have died. It will restart automatically.
把cifar10_test_lmdb、cifar10_train_lmdb及里面的文件都777开放权限了
在迭代的时候也会卡死,,修改迭代次数为2000
把cifar10_test_lmdb、cifar10_train_lmdb及里面的文件都777开放权限了
在迭代的时候也会卡死,,修改迭代次数为2000
阅读全文
0 0
- Caffe学习系列(19): 绘制loss和accuracy曲线
- Caffe学习系列(19): 绘制loss和accuracy曲线
- Caffe学习系列(19): 绘制loss和accuracy曲线
- Caffe学习系列(19): 绘制loss和accuracy曲线
- Caffe学习系列: 绘制loss和accuracy曲线
- caffe 绘制accuracy和loss曲线
- caffe绘制loss和accuracy曲线
- caffe 绘制loss/ accuracy曲线
- caffe绘制loss,accuracy曲线
- caffe学习 绘制loss和accuracy
- Caffe学习:绘制loss和accuracy曲线(使用caffe工具包)
- Caffe学习:绘制loss和accuracy曲线(使用caffe的python接口)
- Caffe学习:使用pycaffe绘制loss、accuracy曲线
- caffe 学习笔记之ubuntu下绘制loss&accuracy 曲线
- Caffe学习:使用pycaffe绘制loss、accuracy曲线
- 绘制loss和accuracy曲线
- caffe绘制训练过程的loss和accuracy曲线
- caffe绘制训练过程的loss和accuracy曲线
- 数据结构实验之查找六:顺序查找
- 使用fileUpload实现上传下载demo
- 欢迎使用CSDN-markdown编辑器
- 程序员面试宝典随笔记(一)--内存管理详解
- NEUQACM OJ 1552: Your name
- Caffe学习系列(19): 绘制loss和accuracy曲线
- 网络调试助手连接远程服务器
- 牛客算法-第二章
- Oracle操作Session的方法
- Effective C++总结
- 关于mapreduce运行错误
- StartActivityForResult方法和finish方法的一些理解
- bzoj4554 [Tjoi2016&Heoi2016]游戏(二分图最大匹配)
- c入门实训