caffe 概述
优质
小牛编辑
121浏览
2023-12-01
caffe 核心
- 数据：带标签的数据集，分为训练集和测试集
- 模型：描述CNN结构的有向无环图，表示对原始数据的处理方式。Caffe模型需要两个重要的参数文件：
- 网络模型(*.prototxt)：定义了网络的每一层行为
- 参数模型(*.solver.prototxt)：定义了网络模型训练过程中需要设置的参数
数据处理
- 原始数据：caffe并不是直接处理原始数据的，而是由预处理程序将原始数据变换存储为LMDB格式，这种方式可以保持较高的IO效率，加快训练时的数据加载速度。
- 模型数据：常用ProtoBuffer文本格式表述，训练结果保存为ProtoBuffer二进制文件或者HDF5格式文件。
编译
编译caffe
make all
make test
make runtest
编译Python wrapper
make pycaffe
编译Matlab wrapper
make matcaffe
工具
./build/tools/caffe
commands:
- train: 训练或者微调一个模型
- test: 对一个模型打分
- device_query: 显示GPU诊断信息
- time: 评估模型执行时间
nvidia显卡
从运算速率看，GTX1080 比 GTX1070 快 25% 左右，GTX1070 稍快于 GTX Titan X，大约是 GTX 960 的三倍。