caffe里增加一个数据预处理层

Main Theme 2017-03-21 10:57:39
我想提取LEVELDB格式的图像文件的中值滤波残差(MFR),即中值滤波后的图像减去原图像。虽然这一步实现跟caffe里均值文件类似只是功能不同,但还是想把这一步实现为一个层。请问foward和backup里还需要添加操作么?如果用GPU写.cu文件的话在原cpp文件的基础上要改动很大么?ps:没学过gpu编程是不是可以放弃了...
...全文
1158 1 打赏 收藏 转发到动态 举报
写回复
用AI写文章
1 条回复
切换为时间正序
请发表友善的回复…
发表回复
beiyechuan1988 2018-12-05
  • 打赏
  • 举报
回复
如果训练需要GPU,需要添加.cpp .hpp .cu 文件,其实里面的东西都是C++,既然有类似的文件,你可以模仿一下,添加完之后,重新编译,就可以使用了。

1,178

社区成员

发帖
与我相关
我的任务
社区描述
C++ Builder 数据库及相关技术
社区管理员
  • 数据库及相关技术社区
加入社区
  • 近7日
  • 近30日
  • 至今
社区公告
暂无公告

试试用AI创作助手写篇文章吧