1. 簡介
Caffe(卷積神經(jīng)網(wǎng)絡(luò)框架)是一個(gè)清晰而高效的深度學(xué)習(xí)框架,純粹的C++/CUDA架構(gòu),支持命令行、Python和MATLAB接口,可以在CPU和GPU直接無縫切換。
2. Caffe的主要優(yōu)勢(shì)
- CPU與GPU的無縫切換;
- 模型與優(yōu)化都是通過配置文件來設(shè)置,無需代碼;
3.Caffe特點(diǎn)
- 上手快:模型與相應(yīng)優(yōu)化都是以文本形式而非代碼形式給出。Caffe給出了模型的定義、最優(yōu)化設(shè)置以及預(yù)訓(xùn)練的權(quán)重,方便立即上手。
- 能夠運(yùn)行最棒的模型與海量的數(shù)據(jù)。Caffe與cuDNN結(jié)合使用,測(cè)試AlexNet模型,在K40上處理每張圖片只需要1.17ms.
- 模塊化:方便擴(kuò)展到新的任務(wù)和設(shè)置上。可以使用Caffe提供的各層類型來定義自己的模型。
- 開放性:公開的代碼和參考模型用于再現(xiàn)。
- 社區(qū)好:可以通過開源社區(qū)和Github參與討論和開發(fā)。
4. Caffe的使用接口
- 命令行
- python
- matlab
5. Caffe官網(wǎng)
Caffe官網(wǎng):http://caffe.berkeleyvision.org/