caffe的学习之路－－－Blob的基本用法

首先，参考书籍《２１天实战caffe》

以前没有Ｃ＋＋基础，所以第一次看这本书时候觉得好垃圾；

现在有了Ｃ＋＋基础，再次看这本书，发现还是很值得学习。

在本文之前，我已经把官方tutorial看完。

要看懂blob.hpp建议将<C++ Primer>前七章看完，再学习caffe源码。

如果要看layer的相关文件，至少把前十五章看完，一起加油。

所以现在开始读源码，首先还是类模板Blob。

下面是自己将Ｂｌｏｂ写入磁盘，从磁盘读取内容存入Blob的程序，当然《２１天实战caffe》的源码。主要写给自己看。

首先先说几个Ｍｅｍｂｅｒ　Ｆｕｎｃｔｉｏｎ

１　 ｖoid Reshape(const int num,const int channels,const int height,const int weight)

      void Reshape(const vector<int> &shape)

这２个函数的作用是重构blob的维数(Dim)

２     void FromProto(const BlobProto &proto, bool reshape=true)

      用法a.FromProto　将Ｂｌｏｂ类型的变量a写入一个类型为ＢｌｏｂＰｒｏｔｏ的变量proto中

　　这是数据从Ｂｌｏｂ到磁盘的中间处理层（作用就是序列化）

3    void ToProto(BlobProto *proto, bool write_diff=fasle )　

     第二个参数输入true　（将diff也序列化，默认是只把data序列化）

另外需要认识２个ＩＯ函数

１.　void WriteProToBinaryFile(const Message &proto,const char *filename)

      void WriteProToBinaryFile(const Message &proto,const string filename)

　　从名字就看出是将ＢｌｏｂＰｒｏ写入二进制文件filename

2.  bool ReadProtoFromBinaryFile(const char *filename,Message *proto)

     bool ReadProtoFromBinaryFile(const string &filename,Message *proto)

#include <vector>#include <iostream>#include <caffe/blob.hpp>#include <caffe/util/io.hpp>using namespace caffe;using namespace std;int main(){    Blob<float> a;    a.Reshape(1,2,3,4);    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()-i-1;    }    a.Update();    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;                }            }        }    }    BlobProto bp1;    a.ToProto(&bp1,true);    WriteProtoToBinaryFile(bp1,"a.blob");    BlobProto bp2;    ReadProtoFromBinaryFile("a.blob",&bp2);    Blob<float> b;    b.FromProto(bp2,true);    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;}