Caffe 的万丈高楼(Net)是按照我们的设计图纸(prototxt),用Blob这些砖块筑程一层层(Layer)楼房,最后通过SGD方法(Solver)进行简装修(Train),精装修(Finetune)实现的。
Caffe使用成为Blob的4维数组用于存储和交换数据。Blob提供了统一的存储器接口,持有一批图像或者其他数据,权值,权值更新值。Blob在内存中表示4维数组,维度从低到高为(width_,height_,channels_,num_).
<span style="color:#333333;">#include <vector>
#include <iostream>
#include <caffe/blob.hpp>
using namespace caffe;
using namespace std;
int main(void)
{
Blob<float> a;
cout<<"Size :"<<a.shape_string()<<endl;
a.Reshape(1,2,3,4);
cout<<"Size :"<<a.shape_string()<<endl;
float *p = a.mutable_cpu_data();
for(int i = 0; i < a.count(); i ++)
{
p[i] = i;
}
for(int u = 0; u < a.num(); u ++)
{
for(int v = 0; v < a.channels(); v ++)
{
for(int w = 0; w < a.height(); w ++)
{
for(int x = 0; x < a.width(); x ++)
{
cout<<"a["<<u<<"]["<< v << "][" << w << "][" << x << "] =" << a.data_at(u,v,w,x)<< endl;
}
}
}
}
return 0;
}
</span>
编译:
1. g++ -o app blob_demo.cpp -I ../include/ -D CPU_ONLY -I ../.build_release/src/ -L ../build/lib/ -lcaffe -lglog -lboost_system
</span>
2. export LD_LIBRARY_PATH=../build/lib/:$LD_LIBRARY_PATH</span>
Blob 文件相关代码相关依赖关系
1. src/caffe/proto/caffe.proto 关于BlobProto
2. 头文件 include/caffe/blob.hpp + include/caffe/syncedmem.hpp (CPU/GPU共享内存类,用于数据同步)
3. 源文件 src/caffe/blob.cpp