Caffe2填坑系列
本专栏的内容是我在使用Caffe2的过程中遇到的问题及解决方法。分享给大家,希望对刚刚使用Caffe2的小伙伴有所帮助。但是我使用的Caffe2是未合并进Pytorch的版本,请您注意。
Sunshine_in_Moon
对计算机视觉,深度学习方向很感兴趣,愿意和大家分享。现在学习大规模人脸识别,人脸检测等相关工作,希望得到大家的帮助!
展开
-
Caffe2填坑系列(7)----make时“cannot find -lopencv_dep_cudart”
在CmakeList.txt 中OpenCV之前,加上set(CUDA_USE_STATIC_CUDA_RUNTIME OFF)或则在make时make -D CUDA_USE_STATIC_CUDA_RUNTIME=OFF原创 2019-05-08 23:18:16 · 498 阅读 · 0 评论 -
Caffe2填坑系列(6)----gcc编译报错:程序中有游离的“\357”、"\273"、"\277"等
原因一:程序中使用了中文的标点符号程序(*.c,*.h)中使用了中文的标点符号(全角),比如;,},+。改成英文的标点半角符号就行了。甚至有时候空格也会出现类似错误,删掉该空格 重新输入。vim里面做类似替换还是很容易的。如何看到报错的符号?od -c hello.c > log.txt在log中就能看到符号了原因二:文件存储格式有问题如果替换成了英文标点还出错的话,还...原创 2019-05-05 23:28:31 · 319 阅读 · 0 评论 -
Caffe2填坑系列(5)----获取输入输出数据----CPU与GPU稍有不同
CPUconst auto& a = Input(0)获取第一个输入给a,a的类型是Tensorauto* b = Output(0)获取第一个输入的指针赋给b获取指向Tensor内数据的指针:const auto* a1 = a.template data(),需要注意两点1.data()前的template ;2、输入一般是只读,我们不希望改变起内部数据,data()返回的是...原创 2019-05-02 12:03:01 · 461 阅读 · 0 评论 -
Caffe2填坑系列(4)----指定梯度计算时的输入和输出
class GetInterpGradient : public GradientMakerBase{ using GradientMakerBase::GradientMakerBase; vector<OperatorDef> GetGradientDefs() override{ return SingleGradientDef( ...原创 2019-04-29 23:43:56 · 324 阅读 · 0 评论 -
Caffe2填坑系列(3)----OPERATOR_SCHEMA()
operator_schema.hNumInputs()// Functions to set the property of the operator schemas. // Sets the number of inputs, either a fixed number or a min and a max. /** * @brief A single in...原创 2019-04-25 23:34:41 · 265 阅读 · 0 评论 -
Caffe2填坑系列(2)----获取数据或参数的方法
pad_end_(OperatorBase::template GetSingleArgument(“pad_end”,0))如果参数“pad_end”没有提供,则采用0为默认值获取Op的输入和输出const auto& X = Input(0):获取第一个输入,注意输入一般为常量类型,因为我们一般只读不更改数据。auto* Y = Output(0): 指向第一个输出,不能为常...原创 2019-04-24 22:23:21 · 284 阅读 · 0 评论 -
Caffe2填坑系列(1)----三个头文件
已经有将近一年的时间没有写博客了。过去的一年了,学习了很多新的东西,在这里和大家分享一下,希望对你有所帮助。特别声明:我在这个系列中所使用的Caffe2,不是已经合并进Pytorch的Caffe2,而是为合并前的。这两者的差别还是很大的。今天,作为第一讲,内容可以说非常少。- #include "caffe2/core/context.h"选择是CPU或者GPU- #include "...原创 2019-04-19 23:29:18 · 328 阅读 · 0 评论 -
Caffe To Caffe2 Python脚本caffe_translator.py详细注释
将Caffe的模型转换成Caffe2模型,官方提供了一个caffe_translator.py脚本。在此我做了详细的注释。## @package caffe_translator# Module caffe2.python.caffe_translator#!/usr/bin/env python2import argparseimport copyimport loggingim...原创 2019-01-06 11:20:53 · 969 阅读 · 0 评论