随笔分类 -  caffe

摘要:import numpy as np import caffe import os f = open('/home/python_code/result.txt', 'wr') caffe.set_mode_gpu() net = caffe.Net('/home/python_code/caffe_predict/deploy.prototxt', '/ho... 阅读全文
posted @ 2016-12-11 16:08 拖拉机的音响 阅读(2503) 评论(1) 推荐(0)
摘要:在训练网络时可以利用别人的pre-train model来初始化的网络,caffe可以实现两个网络参数的转化,前提条件是转化的层的参数设计是一致的,以下程序是转化了三个卷积层和三个全连接层的参数,python的代码如下: 阅读全文
posted @ 2016-12-11 15:58 拖拉机的音响 阅读(950) 评论(0) 推荐(0)
摘要:step1:将图片数据转化为lmdb格式 caffe中建议将图片转化为lmdb格式,我们可以利用caffe提供的工具中的convert_imageset,这个工具是需要使用open cv,因此在配置caffe中需要注意。我使用的OPENCV的2.4。 ## Refer to http://caffe 阅读全文
posted @ 2016-11-08 20:11 拖拉机的音响 阅读(1188) 评论(0) 推荐(0)
摘要:caffe的model是需要用prototxt来定义的,如果要训练自己的model我们就需要自己来写prototxt文件,这篇博文主要写caffe的layer是如何来定义的。 参考:http://caffe.berkeleyvision.org/tutorial/layers.html layer的 阅读全文
posted @ 2016-11-06 19:46 拖拉机的音响 阅读(6031) 评论(0) 推荐(0)