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

最新推荐文章于 2021-06-27 16:06:29 发布

原创最新推荐文章于 2021-06-27 16:06:29 发布 · 4.4k 阅读

4 ·

CC 4.0 BY-SA版权

caffe学习之路专栏收录该内容

5 篇文章

订阅专栏

本文详细介绍Caffe中Blob类的使用方法，并提供从磁盘读取和写入Blob的具体实现示例。适合有一定C++基础的学习者进一步理解Caffe框架。

首先，参考书籍《２１天实战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;

}