您的位置：首页 > 理论基础 > 数据结构算法

21天实战caff--2 第8天 Caffe数据结构-Blob基本用法

2016-08-20 13:18 387 查看

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

内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理

标签：

相关文章推荐

新的分享

章节导航