- 深度學習與圖像識別:原理與實踐
- 魏溪含 涂銘 張修鵬
- 425字
- 2019-08-15 18:06:35
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/