Caffe
-
caffe的python接口学习（3）训练模型training
如果不进行可视化，只想得到一个最终的训练model, 那么代码非常简单，如下 : import caffe caffe.set_device(0) caffe.set_mode_gpu() solver = caffe.SGDSolver(‘/home/xxx/data/solver.prototxt’) solver.solve()
-
caffe的python接口学习（6）用训练好的模型caffemodel分类新图片
经过前面两篇博文的学习，我们已经训练好了一个caffemodel模型，并生成了一个deploy.prototxt文件，现在我们就利用这两个文件来对一个新的图片进行分类预测。 我们从mnist数据集的test集中随便找一张图片，用来进行实验。 #coding=utf-8 import caffe import numpy as np root=’/home/x…
-
caffe报错：cudnn.hpp:86] Check failed: status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM 原因
在实际项目中出现的该问题，起初以为是cudnn版本的问题，后来才定位到在网络进行reshape操作的时候 input_layer->Reshape({(int)imgin.size(), input_layer->shape(1), input_layer->shape(2), input_layer->shape(3)}); 如上所…
-
caffe中的BatchNorm层
在训练一个小的分类网络时，发现加上BatchNorm层之后的检索效果相对于之前，效果会有提升，因此将该网络结构记录在这里，供以后查阅使用： 添加该层之前： 1 layer { 2 name: “conv1” 3 type: “Convolution” 4 bottom: “data” 5 top: “conv1” 6 param { 7 lr_mult: 1…
-
caffe中的卷积
https://www.zhihu.com/question/28385679 如上，将三维的操作转换到二维上面去做，然后调用GEMM库进行矩阵间的运算得到最后结果。 两个矩阵相乘，需要中间的那个维度相同，这个相同的维度就是C×K×K，其中C是Feature map的维度，K为卷积核的边长。 按照卷积核在Feature map上面滑窗的顺序将其展开成…
-
caffe中在某一层获得迭代次数的方法以及caffe编译时报错 error: ‘to_string’ is not a member of ‘std’解决方法
https://stackoverflow.com/questions/38369565/how-to-get-learning-rate-or-iteration-times-when-define-new-layer-in-caffe 参考上述网址上的方法，需要修改 common.hpp class Caffe { public: static Caff…
-
caffe编译报错 cudnn.hpp:127:41: error: too few arguments to function ‘cudnnStatus_t cudnnSetPooling2dDescriptor
转载自： https://blog.csdn.net/u011070171/article/details/52292680 这是因为当前版本的caffe的cudnn实现与系统所安装的cudnn的版本不一致引起的。 解决办法： 1.将./include/caffe/util/cudnn.hpp 换成最新版的caffe里的cudnn的实现，即相应的cudn…
-
cuda9.0编译caffe报错nvcc fatal : Unsupported gpu architecture ‘compute_70’
Tesla V100 cuda9.0 caffe编译的时候报上述错误，修改方法： CUDA_ARCH := #-gencode arch=compute_20,code=sm_20 \ #-gencode arch=compute_20,code=sm_21 \ -gencode arch=compute_30,code=sm_30 \ -gencode a…
-
Chapter 1 初探Caffe
首先下载windows下源码： Microsoft 官方：GitHub – Microsoft/caffe: Caffe on both Linux and Windows 官方源码使用Visual Studio 2013工程，使用vs2013及以上版本均可以打开，但是需要VS2013编译环境，所以得安装VS2013。 打开caffe-masterwindo…
-
如何使用 Opencv dnn 模块调用 Caffe 预训练模型？
QString modelPrototxt = “D:\\Qt\\qmake\\CaffeModelTest\\caffe\\lenet.prototxt”; QString modelBin = “D:\\Qt\\qmake\\CaffeModelTest\\caffe\\snapshot_iter_10000.caffemodel”; QString i…
