Blob进阶4


Blob在内存中表示四维数组,维度由高到底为(num_,channels_,height_,width_),

num:第几帧,存储数据或权值(data)和权值增量(diff) { 相当重要的两个概念}

channels:颜色通道RGB

height:图像的高

width:图像的宽

Blob是一个模板类,创建时需要制定模板参数,建一个Blob_demo.cpp程序放在caffe根目录:


通过ToProto()将Blob值保存到磁盘

通过FromProto()从磁盘载入内存


代码用的WriteProtoToBinaryFile 与ReadProtoFromBinaryFileOrDie 在include/caffe/util声明 在src/caffe/util/io.cpp定义 点击打开链接

#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);   //调用Reshape赋值
 cout<<"Size : "<<a.shape_string()<<endl;
  float *p=a.mutable_cpu_data(); //mutable_cpu_data()函数修改数值
  float *q=a.mutable_cpu_diff(); //diff权值增量,源码频繁出现
 for(int i=0;i<a.count();i++)
  {
    p[i]=i;      //把data初始化为0,1,2,3
    q[i]=a.count()-1-i;//把diff初始化为23,22,21.。。      }
   }
  a.Update(); //此Update操作将data与diff融合,是权值更新最终实施者 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<<"b["<<u<<"]["<<v<<"]["<<w<<"]["<<x<<"]="<<a.data_at(u,v,w,x)<<endl;
		    }
		  }
		}
	     }
         cout<<"ASUM="<<a.asum_data()<<endl;  //计算所以元素的L1范数 (绝对值之和)
         cout<<"SUMSQ="<<a.sumsq_data()<<endl;//计算所以元素的L2范数(平方和)

BlobProto bp;    //构造一个BlobProto对象
a.ToProto(&bp,true);  //将a序列化,连同diff(默认不带)
WriteProtoToBinaryFile(bp,"a.blob");     //写入磁盘文件 
BlobProto bp2; //构造一个新的BlobProto对象
ReadProtoFromBinaryFileOrDie("a.blob",&bp2); //读入磁盘文件 函数解析 
Blob<float> b; //新建一个Blob对象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++)
                  {
                         //打印b
                    cout<<"a["<<u<<"]["<<v<<"]["<<w<<"]["<<x<<"]="<<b.data_at(u,v,w,x)<<endl;
		    }
		  }
		}
	     }
         cout<<"ASUM="<<b.asum_data()<<endl;  //计算所以元素的L1范数 (绝对值之和)
         cout<<"SUMSQ="<<b.sumsq_data()<<endl;//计算所以元素的L2范数(平方和)
        }

用如下命令编译(之前需要编译好caffe),路径你要改的:

g++ -o app blob_demo.cpp -I/home/hk/Documents/caffe-master/include/ -D CPU_ONLY -I/home/hk/Documents/caffe-master/.build_release/src/ -L/home/hk/Documents/caffe-master/build/lib/ -lcaffe -lglog -lboost_system

会发现根目录下生成了app文件


接着运行:(路径你要改的)

  1. <span style="font-size:18px;"> export LD_LIBRARY_PATH=/home/hk/Documents/caffe-master/build/lib/:$LD_LIBRARY_PATH  
  2. </span>  

  1. ./app  


结果:



b克隆成功!

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值