Sunshine_in_Moon的专栏

天行健君子以自强不息!

Caffe2填坑系列(10)----编译成功,在python中使用时报”key already registered. Offending key: ImageInput“

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

2019-05-20 23:11:22

阅读数 23

评论数 0

Caffe2填坑系列(9)----错误提示“undefined reference to caffe2::CUDAContext::“

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

2019-05-14 23:50:12

阅读数 24

评论数 0

Caffe2填坑系列(8)----获得Op输入和输出blob的个数的两种方法

构造函数中,任何一个Op都是公有继承Operator,构造函数一般是 C++ :Operator<Context>(def,ws), … 可以看到每一个构造函数里都有一个OperatorDef的对象def,OperatorDef在protot里进行定义,主要时组着里...

2019-05-12 22:22:34

阅读数 20

评论数 0

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

阅读数 58

评论数 0

Caffe2填坑系列(6)----gcc编译报错:程序中有游离的“\357”、"\273"、"\277"等

原因一:程序中使用了中文的标点符号 程序(*.c,*.h)中使用了中文的标点符号(全角),比如;,},+。改成英文的标点半角符号就行了。甚至有时候空格也会出现类似错误,删掉该空格 重新输入。 vim里面做类似替换还是很容易的。 如何看到报错的符号? od -c hello.c > ...

2019-05-05 23:28:31

阅读数 28

评论数 0

Caffe2填坑系列(5)----获取输入输出数据----CPU与GPU稍有不同

CPU const auto& a = Input(0)获取第一个输入给a,a的类型是Tensor auto* b = Output(0)获取第一个输入的指针赋给b 获取指向Tensor内数据的指针: const auto* a1 = a.template data(),需要注意两...

2019-05-02 12:03:01

阅读数 30

评论数 0

提示
确定要删除当前文章?
取消 删除
关闭
关闭