Caffe数据结构一Blob

Blob是Caffe的基本存储单元，用于存储权值（data）和权值增量（diff）；

BlobProto对象实现磁盘，内存之间的数据通信。对于加载，保存训练好的模型的权值非常实用。

#include <vector>
#include <iostream>
#include <caffe/blob.hpp>
#include <caffe/util/io.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();
  float * q = a.mutable_cpu_diff();//权值增量
  for(int i = 0; i < a.count(); i++)
  {
    p[i] = i;
    q[i] = a.count() - 1 - i;
  }
  a.Update(); //更新权值 data =data -diff
  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;
        }
      }
    }
  }
  cout<<"ASUM = "<<a.asum_data()<<endl;  //求一范数
  cout<<"SUMSQ = "<<a.sumsq_data()<<endl;//二范数，平方和
  BlobProto bp;  //构造对象
  a.ToProto(&bp, true); //将a序列化，两桶diff（默认不带）
  WriteProtoToBinaryFile(bp, "a.blob");  //写入磁盘文件
  BlobProto bp2;
  ReadProtoFromBinaryFileOrDie("a.blob", &bp2);//读取磁盘文件
  Blob<float> b;
  b.FromProto(bp2, true);//从序列化对象bp2中克隆b
  for(int u = 0; u < b.num(); u++)
  {
    for(int v = 0; v < b.channels(); v++)
    {
      for(int w = 0; w < b.height(); w++)
      {
        for(int x = 0; x < b.width(); x++)
        {
          cout<<"b["<<u<<"]["<<v<<"]["<<w<<"]["<<x<<"] = "<< b.data_at(u, v, w, x)<<endl;
        }
      }
    }
  }
  return 0;
}

g++ -o app blob_demo_proto.cpp  -I /home/zhao/ming/caffe0/caffe/include/ -D CPU_ONLY -I /home/zhao/ming/caffe0/caffe/.build_release/src/ -L /home/zhao/ming/caffe0/caffe/build/lib/ -lcaffe -lglog -lboost_system

   在源文件的同目录下会产生 a.blob  app  blob_demo.cpp