在caffe中增加和convolution相同的层

1.打开vision_layers.hpp,复制ConvolutionLayer的代码,把类名还有构造函数的名字改为WtfLayer,把里面的带GPU的函数删掉.

2.Wtf_layer.cpp 添加到src\caffe\layers文件夹中,代码内容复制conv_layer.cpp 把对应的类名修改

3.修改proto/caffe.proto文件,在optional ConvolutionParameter convolution_param = 106;的下一行添加

                                                  optional WtfParameter wtf_param = 141;

                                               在optional ConvolutionParameter convolution_param = 10; 的下一行添加

                                                  optional WtfParameter wtf_param = 43;

     添加和message ConvolutionParameter一样的代码

4.在layer_factory.cpp中添加和GetConvolutionLaye里面代码一样的代码,需要改相应名字。

posted @   Shiyu_Huang  阅读(300)  评论(0编辑  收藏  举报
编辑推荐:
· 开发者必知的日志记录最佳实践
· SQL Server 2025 AI相关能力初探
· Linux系列:如何用 C#调用 C方法造成内存泄露
· AI与.NET技术实操系列(二):开始使用ML.NET
· 记一次.NET内存居高不下排查解决与启示
阅读排行:
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
点击右上角即可分享
微信分享提示