美文网首页
STNcaffe移植过程

STNcaffe移植过程

作者: 帅气的我要加油 | 来源:发表于2018-04-09 13:55 被阅读6次

看到自己的添加stn的lstm-caffe跑起来简直要激动的哭了,记录一下移植过程,参考:
https://blog.csdn.net/kuaitoukid/article/details/51035028
stn的git代码:
https://github.com/happynear/SpatialTransformerLayer

1.首先,编译一个全新的caffe-master。第一个链接中下载的有3个文件夹:examples、include和src

2.将STN/include/caffe中的.hpp(除filler.hpp)复制到 caffe-master/include/caffe/layers/ 中;
将.hpp中的 #include "caffe/neuron_layers.hpp" 改为 #include "caffe/layers/loss_layer.hpp";

3.将STN/include/caffe/filler.hpp中修改的部分加入到caffe-master/include/caffe/filler.hpp 中

4.将STN/src/caffe/layers/ 中的文件复制到 caffe-master/src/caffe/layers/ 中;
loc_loss_layer.cpp:
#include "caffe/vision_layers.hpp" -> //#include "caffe/vision_layers.hpp"
#include "caffe/loc_loss_layer.hpp" -> #include "caffe/layers/loc_loss_layer.hpp"
power_file_layer.cpp:
#include "caffe/power_file_layer.hpp" -> #include "caffe/layers/power_file_layer.hpp"
st_layer.cpp:#include "caffe/st_layer.hpp" -> #include "caffe/layers/st_layer.hpp"
st_loss_layer.cpp:
#include "caffe/vision_layers.hpp" -> //#include "caffe/vision_layers.hpp"
#include "caffe/st_loss_layer.hpp" -> #include "caffe/layers/st_loss_layer.hpp"
.cu 和 .cpp 做相同的修改

5.将STN/src/caffe/proto/caffe.proto 中修改的部分加入到caffe-master/src/caffe/proto/caffe.proto 中,添加的时候一定要仔细

最后一步编译就好啦

相关文章

  • STNcaffe移植过程

    看到自己的添加stn的lstm-caffe跑起来简直要激动的哭了,记录一下移植过程,参考:https://blog...

  • 华为鲲鹏专家解读:90%代码如何移植到鲲鹏平台

    摘要:探讨一下软件移植到鲲鹏平台过程的原理,以及软件工程的相应的过程。 Linux环境下跨平台软件移植过程中,需要...

  • 试验品(目录)

    作品简介:2088年,人类和人工智能竞争过程中,人类制造PS芯片,移植大脑,改变人类基因,PS芯片移植大脑过程的初...

  • 为什么大树移植时土球包扎物不解现象普遍存在,到底该不该解?

    在大树移植的过程中,除了极少数的小规格树木品种可以裸根移植外,大部分树木的移植都是需要带土球的。带土球进行移植的树...

  • 移植openssh

    概述 本文讲述移植openssh的过程。移植环境:嵌入式操作系统:Linux-3.18.20交叉编译器:arm-l...

  • 2019-01-08

    妈的,出科就有肝移植和肾移植,真棒,真有天理。跟男票讲了死人的全过程,累了。明天真的会更好吗?

  • Linphone-移植(二)

    这一次的博客主要是对于自己移植一些过程进行记录,第一次移植linphone就没有做记录,以至于在新公司的第二次移植...

  • 阴毛种植注意事项有哪些

    阴毛种植注意事项有哪些?阴毛移植术过程中是没有痛苦的,会在治疗部位注射局部麻醉,并且阴毛移植效果是永久的,不会产生...

  • Jetson Xavier NX 内核编译移植

    移植过程参考:JetsonNX快速入门指南[https://docs.nvidia.com/jetson/l4t/...

  • 华为实验-移植程序测试过程

    实验六 实验任务配置 步骤1 移植光敏传感器代码,通过传感器采集光照亮度 在keil环境中右击“IoT-Demo...

网友评论

      本文标题:STNcaffe移植过程

      本文链接:https://www.haomeiwen.com/subject/hkludxtx.html