Sunshine_in_Moon的专栏

将Caffe的模型转换成Caffe2模型，官方提供了一个caffe_translator.py脚本。在此我做了详细的注释。## @package caffe_translator# Module caffe2.python.caffe_translator#!/usr/bin/env python2import argparseimport copyimport loggingim...

原因：ImageInput这个Op被注册了多次，我在编写自己的.cc文件时，是以image_input.cc为模板，前面我都改成了自己Op的名字，包括REGISTER_CPU_OPERATOR(),OPERATOR_SCHEMA().唯独没有改NO_GRADIENT()，造成了上述错误。具体代码可参考core/register.h...

一般出现”undefined reference to…“的错误是找不到相应的库造成的，但我自己在编写完一个Op后编译时出现了这个问题，经过一天的查找终于找到了根源所在，知道原因后自己都无语了，原来在caffe2中文件的命名是有规则了，不能胡乱起名。我错将_op_gpu.cc,写成了_gpu_op.cc首先，看一下在image/CMakeLists.txt中的设置if(USE_OPENCV ...

在CmakeList.txt 中OpenCV之前，加上set(CUDA_USE_STATIC_CUDA_RUNTIME OFF)或则在make时make -D CUDA_USE_STATIC_CUDA_RUNTIME=OFF

原因一：程序中使用了中文的标点符号程序（＊.c，＊.h）中使用了中文的标点符号(全角)，比如；，｝，＋。改成英文的标点半角符号就行了。甚至有时候空格也会出现类似错误，删掉该空格 重新输入。vim里面做类似替换还是很容易的。如何看到报错的符号？od -c hello.c > log.txt在log中就能看到符号了原因二：文件存储格式有问题如果替换成了英文标点还出错的话，还...

class GetInterpGradient ： public GradientMakerBase{ using GradientMakerBase::GradientMakerBase; vector<OperatorDef> GetGradientDefs() override{ return SingleGradientDef( ...

pad_end_(OperatorBase::template GetSingleArgument(“pad_end”,0))如果参数“pad_end”没有提供，则采用0为默认值获取Op的输入和输出const auto& X = Input(0)：获取第一个输入，注意输入一般为常量类型，因为我们一般只读不更改数据。auto* Y = Output(0): 指向第一个输出，不能为常...

已经有将近一年的时间没有写博客了。过去的一年了，学习了很多新的东西，在这里和大家分享一下，希望对你有所帮助。特别声明：我在这个系列中所使用的Caffe2，不是已经合并进Pytorch的Caffe2，而是为合并前的。这两者的差别还是很大的。今天，作为第一讲，内容可以说非常少。- #include "caffe2/core/context.h"选择是CPU或者GPU- #include "...

CPUconst auto& a = Input(0)获取第一个输入给a，a的类型是Tensorauto* b = Output(0)获取第一个输入的指针赋给b获取指向Tensor内数据的指针：const auto* a1 = a.template data(),需要注意两点1.data()前的template ;2、输入一般是只读，我们不希望改变起内部数据，data()返回的是...

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...

构造函数中，任何一个Op都是公有继承Operator，构造函数一般是 C++:Operator<Context>(def,ws),…可以看到每一个构造函数里都有一个OperatorDef的对象def,OperatorDef在protot里进行定义，主要时组着里一个Op的一些信息，我们可以调用def.input_size()或def.output_size()获得输入输出的...