caffe
qq649185776
这个作者很懒,什么都没留下…
展开
-
利用python接口进行单张图片测试
import numpy as npimport syscaffe_root = '/home/tyd/caffe/' #the file should be run from (caffe_root)/examples sys.path.insert(0,caffe_root + 'python')import caffecaffe.set_mode_cpu()model_de...原创 2019-11-10 12:13:41 · 392 阅读 · 0 评论 -
利用自定义python层进行定义网络
在prototxt文件中配置layer层layer{ name:'MyPythonLayer' type:'Python' top:'output' bottom:'conv' python_param{ module:'mypythonlayer' layer:'MyLayer' param_str:"'num':21"}定义python程序,类名与protot...原创 2019-11-10 12:10:00 · 352 阅读 · 1 评论 -
利用python接口生成prototxt文件
import numpy as npimport matplotlib.pyplot as pltimport sys,oscaffe_root = '/home/tyd/caffe/'sys.path.insert(0,caffe_root+'python')from caffe import layers as Lfrom caffe import params as Pimp...原创 2019-11-10 12:04:36 · 362 阅读 · 0 评论 -
利用python接口进行训练caffe网络
import numpy as npimport matplotlib.pyplot as pltimport sys,oscaffe_root = '/home/tyd/caffe/'sys.path.insert(0,caffe_root+'python')import caffe#caffe.set_device(0)caffe.set_mode_cpu()solver ...原创 2019-11-10 12:03:04 · 282 阅读 · 0 评论 -
生成h5文件
生成训练h5文件import h5pyimport osimport cv2import mathimport numpy as npimport randomimport root_path = "/home/tyd/caffe_case/HDF5/image"with open("/home/tyd/caffe_case/HDF5/hdf5.txt","r") ...原创 2019-11-10 12:01:47 · 3162 阅读 · 1 评论 -
caffe训练网络
训练网络#sudo sh ./build/tools/caffe train --solver=examples/mnist/train_lenet.sh-solver:必选参数。一个protocol buffer类型的文件,即模型的配置文件。如:# ./build/tools/caffe train -solver examples/mnist/lenet_solver.prototx...原创 2019-11-10 11:59:46 · 214 阅读 · 0 评论 -
caffe绘制网络
计算均值命令sudo /home/tyd/caffe/build/tools/compute_image_mean /home/tyd/caffe/examples/mnist/mnist_train_lmdb /home/tyd/caffe_case/mean.binaryproto绘制网络1.安装graphViz#sudo apt-get install graphviz2...原创 2019-11-10 11:58:06 · 73 阅读 · 0 评论