
caffe
Felaim
Hey, you're not a nobody. You're a somebody.
展开
-
Caffe:常用层的参数设置
caffe常用层参数设置 int K_ int N_ bool bias_term_...原创 2022-12-06 13:25:43 · 211 阅读 · 0 评论 -
Caffe:常见层的书写方式
caffe的常用层书写方式原创 2022-12-06 13:23:04 · 371 阅读 · 0 评论 -
深度学习:模型转换(pth转onnx转caffemodel)
模型转换(pth转onnx转caffemodel)一、背景介绍因为模型要用到前端,所以要将pytorch训练的模型转成caffemodel,之前LZ已经写了一片tensorflow转caffemodel的教程,今天就总结一篇pytorch转onnx转caffemodel的教程二、pth转onnx这个严格意义上是DBFace系列的第四部分,但是主要是用来转模型,对其他模型也是适用的,于是就单独拎出来了,首先放上代码import commonimport eval_toolimport torch原创 2020-08-20 16:15:00 · 6821 阅读 · 2 评论 -
深度学习:一文解决卷积,池化,反卷积的维度问题(包含一些工程思考)
一文解决卷积,池化,反卷积的维度问题1. 背景其实这个之前LZ对这个关注度不是很高,像tensorflow,选择‘valid’和‘same’也就可以了,但是对于caffe不可以。并且在海思芯片使用的caffe1.0,据前端同事说很多骚操作在前端都是不支持的,哭泣中,即使在重训的时候可以重新写C,来添加层,在NNIE推断结果还是不行。。。按理说华为内部应该是有一波大神来构建对应的代码来支持比较新的网络层的,这应该也算是一种技术壁垒吧,纯属猜测。还有量化误差的问题,这个真的也是无力吐槽,据说9月份就会出原创 2020-07-18 16:40:19 · 1309 阅读 · 0 评论 -
Caffe: Eltwise layer 详解 (pycaffe, 卷积层相加\减等操作)
Caffe: Eltwise layer 详解1. eltwise_layer的头文件#ifndef CAFFE_ELTWISE_LAYER_HPP_#define CAFFE_ELTWISE_LAYER_HPP_#include <vector>#include "caffe/blob.hpp"#include "caffe/layer.hpp"#include "...原创 2020-04-29 17:26:32 · 1176 阅读 · 0 评论 -
Caffe: 为什么你的tf模型转caffemodel老不对?(其中一个重要原因)
使用训练好的tensorflow模型,转换成caffemodel的时候,需要对tensorflow中接口设置和caffe中接口设置都非常了解,才能对其参数进行对应设置,举个简单的例子:在我们使用slim库的时候,只需要按照如下方式进行书写:net = slim.conv2d(inputs, 16, 3, stride=1, padding='VALID', scope='Conv1')但是如...原创 2020-03-28 14:52:35 · 545 阅读 · 0 评论 -
Caffe: tensorflow的模型到底怎么转成caffemodel??(完整的流程+代码讲解)
tensorflow的模型到底怎么转成caffemodel?1. 简介一般情况下,LZ是使用tensorflow的,好像之前在学校只要python跑通,benchmark再刷高一点就行了。现在在公司,要把自己的模型进行移植,之前是后端的移植,最近前端也提了需求,前端一般都是用海思芯片,只支持caffe,所以为了先测试时间得把tf的模型转成caffemodel。程序猿总有不会的,总有要学习的不...原创 2020-03-28 14:05:10 · 1386 阅读 · 4 评论 -
Caffe: (三) LeNet (训练测试网络,solver的参数理解,python接口搭建网络结构)
from pylab import *%matplotlib inlineimport caffeimport oscaffe_root = "Path to caffe"os.chdir(caffe_root)os.getcwd()“Path to caffe”!data/mnist/get_mnist.shDownloading...# prepare data...原创 2020-03-05 17:44:44 · 534 阅读 · 0 评论 -
Caffe: (二) Classification (CPU/GPU设置, feature map可视化,层名,参数输出, Top-1、Top-5输出)
#set up python environment: numpy for numerical routines, matplotlib for plottingimport numpy as npimport matplotlib.pyplot as plt#display plots in this notebook%matplotlib inlineplt.rcParams['f...原创 2020-03-04 19:05:09 · 580 阅读 · 0 评论 -
Caffe: (一) Net Surgery (卷积,全连接的weights和biases的参数存储方式理解)
import numpy as npimport matplotlib.pyplot as pltimport sys%matplotlib inlineimport caffecaffe_example_path = "your caffe example path"sys.path.append(caffe_example_path)#configure plottingplt...原创 2020-03-03 20:30:48 · 603 阅读 · 0 评论 -
Caffe: CMake源码编译(OpenCV4.2+CUDA)
周六周日在家倒腾了两天,总算稍微了解点pycaffe的一些接口和api设置了,但是针对不同层,还是有很多注意的细节的,所以也准备写个博客专门总结pycaffe的常见层参数设置,(LZ这个蠢啊,之前居然手写prototxt,一把辛酸泪,太傻了/(ㄒoㄒ)/~~)之前已经写过用Anaconda直接安装caffe,这样安装caffe不香嘛,为啥非得源码再来安装一次呢?1.项目里后端是用的tensor...原创 2020-02-23 22:29:59 · 1576 阅读 · 0 评论 -
Anaconda: 快速安装caffe
临时接到的任务,要在两天时间里,把tensorflow的模型转成caffe模型,但是LZ一上来学习的就是tf,所以对于caffe实在了解不多,但是任务紧急,不管如何,也得硬着头皮学习移植,把CUDA代码构建暂时先放两天,下面就是anaconda配置caffe的过程1.首先确保已经安装好了anaconda2.创建Anaconda对应的caffe envconda create -n caffe...原创 2020-02-18 10:24:35 · 1776 阅读 · 7 评论