caffe框架简化之TinyCaffe

                                 TinyCaffe----一款原生态轻量化caffe前向版本

 

         众所周知,caffe框架的依赖庞大得恐怖:boost占了半壁江山(几百兆),lmdb、levedb,hdf5,nccl等等,原始框架实际部署的时候甚至高达近1GB,但是实际上前向过程仅仅依赖protobuf,函数调用仅仅是forward系列函数,Solver之类的都没有用-----网上诸位大神为此出了形形色色的轻巧性caffe,比较好的是mini-caffe。

       本着能让caffe调用上更加原生态且依赖减少的初衷,本人在仔细梳理caffe框架之后也DIY了一个版本的caffe,和mini-caffe不同的是本人仅仅是去除了一些冗余的依赖,caffe前向调用更加原生态,只是库变小了-------------称之为tiny-caffe。

      简化版本的caffe取自happynear的caffe-windows,因为这个版本的caffe相比BLVC的caffe添加了一些新的层,后续我也会添加相应的层来逐步丰富自己的tiny-caffe

        拖延了很久了.....终于完成了,编译测试通过(windows),ubuntu暂时没有测试,不过ubuntu下只要依赖不缺,不会有windows下那么麻烦!

github: https://github.com/SyGoing/tinycaffe 有任何问题都可以issue,喜欢的话给个star哦~~~~

 

该框架提炼完成也参考了mini-caffe中的设备内存管理那一块的代码:syncedmem.hpp,感谢mini-caffe作者luoyetx

 

评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值