深度学习与图像识别:原理与实践
上QQ阅读APP看书,第一时间看更新

2.1.6 Caffe

Caffe是基于C++语言编写的深度学习框架,作者是中国人贾扬清。它开放源码(具有Licensed BSD),提供了命令行,以及Matlab和Python接口,清晰、可读性强、容易上手。

Caffe是早期深度学习研究者使用的框架,由于很多研究人员在上面进行开发和优化,因此其现今也是流行的框架之一。Caffe也存在不支持多机、跨平台、可扩展性差等问题。初学者使用Caffe时还需要注意下面这些问题。

1)Caffe的安装过程需要大量的依赖库,因此会涉及很多安装版本问题,初学者不易上手。

2)当用户想要实现一个新的层时,需要用C++实现它的前向传播和反向传播代码,而如果想要新层运行在GPU之上,则需要同时使用CUDA实现这一层的前向传播和反向传播。

Caffe2出自Facebook人工智能实验室与应用机器学习团队,但贾扬清仍是主要贡献者之一。Caffe2在工程上做了很多优化,比如运行速度、跨平台、可扩展性等,它可以看作是Caffe更细粒度的重构,但在设计上,其实Caffe2与TensorFlow更像。目前代码已开源。

总结:至今工业界和学界仍有很多人在使用Caffe,而Caffe2的出现为我们提供了更多的选择。

项目地址:Caffe:http://caffe.berkeleyvision.org/

Caffe2:https://caffe2.ai/