caffemodel之庖丁解牛

- 日理万妓 2022-05-23 10:56 205阅读 0赞

很多时候算法工程师把精力更多的聚焦于当下最流行的框架。一旦出现了一些各个式样的流行框架，便迫不急待的在自己的数据集或者公共的数据集上跑。然后生成模型来验证准确率与检测速率。如果以上两者不达标便会修改数据集或者对模型调整参数。导致时下有些段子就说：“优秀的调参侠就是好的算法工程师”。暂且不论这个观点是否正确，很多时候整个深度学习过程就是这样的。

这两天突然很好奇，我用CAFFE生成的caffemodel里面到底是些什么数据。现在有很多方法可以让模型可视化。我在网上搜索了一些方法，并且做也一点实践工作，在此时写下来供大家参考。（下面的有些部分参考了其他人的博客）

**Classifier类的构造函数的时候，有一个net\_指针，并且执行了一个操作：**

**\[cpp\]** [view plain][] [copy][view plain]

1.  net\_->CopyTrainedLayersFrom(trained\_file);

**这个CopyTrainedLayersFrom函数就是从caffemodel里面载入我们需要的参数了。然后笔者就从这个函数开始挖，进入net.cpp文件：**

**\[cpp\]** [view plain][] [copy][view plain]

1.  template <typename Dtype>
2.  void Net<Dtype>::CopyTrainedLayersFrom(const string trained\_filename) \{
3.  if (H5Fis\_hdf5(trained\_filename.c\_str())) \{
4.  CopyTrainedLayersFromHDF5(trained\_filename);
5.  \} else \{
6.  CopyTrainedLayersFromBinaryProto(trained\_filename);
7.  \}
8.  \}

**在这里很明显执行了else下面的语句，从二进制文件中去读取了参数，然后，笔者又找到了CopyTrainedLayersFromBinaryProto函数，很巧就在CopyTrainedLayersFrom函数的下方：**

**\[cpp\]** [view plain][] [copy][view plain]

1.  template <typename Dtype>
2.  void Net<Dtype>::CopyTrainedLayersFromBinaryProto(
3.  const string trained\_filename) \{
4.  NetParameter param;
5.  ReadNetParamsFromBinaryFileOrDie(trained\_filename, &parm);
6.  CopyTrainedLayersFrom(param);
7.  \}

**在里面首先执行了一个ReadNetParamsFromBinaryFileOrDie函数，把二进制文件中的参数读到parm里面，parm是一个NetParameter类型的，NetParameter继承了实际是一个Message，Message是proto类型的，详细的笔者后话解析。然后笔者去找了一下ReadNetParamsFromBinaryFileOrDie函数，这个函数在upgrade\_proto.cpp里面：**

**\[cpp\]** [view plain][] [copy][view plain]

1.  void ReadNetParamsFromBinaryFileOrDie(const string& param\_file,
2.  NetParameter\* param) \{
3.  CHECK(ReadProtoFromBinaryFile(param\_file, param))
4.  << "Failed to parse NetParameter file: " << param\_file;
5.  UpgradeNetAsNeeded(param\_file, param);
6.  \}

**这个函数完成的功能是首先进行ReadProtoFromBinaryFile，然后执行了一个更新的操作。然后笔者就去找这个ReadProtoFromBinaryFile函数，这个函数在io.cpp里面：**

**\[cpp\]** [view plain][] [copy][view plain]

1.  bool ReadProtoFromBinaryFile(const char\* filename, Message\* proto) \{
2.  int fd = open(filename, O\_RDONLY);
3.  CHECK\_NE(fd, -1) << "File not found: " << filename;
4.  ZeroCopyInputStream\* raw\_input = new FileInputStream(fd);
5.  CodedInputStream\* coded\_input = new CodedInputStream(raw\_input);
6.  coded\_input->SetTotalBytesLimit(kProtoReadBytesLimit, 536870912);
7.  
8.  bool success = proto->ParseFromCodedStream(coded\_input);
9.  
10. delete coded\_input;
11. delete raw\_input;
12. close(fd);
13. return success;
14. \}

**这个函数就比较底层了，读者朋友们可以看到，这个函数里面就使用了open函数，和一些底层的google::protobuf的数据流。在这里我们其实就明白，是先把二进制文件(caffemodel)转化成文件流，再放入proto里面。**

**那么，笔者大胆猜想，能读就能写。**

**其实io.cpp里面已经定义了这种接口：**

**\[cpp\]** [view plain][] [copy][view plain]

1.  void WriteProtoToTextFile(const Message& proto, const char\* filename) \{
2.  int fd = open(filename, O\_WRONLY | O\_CREAT | O\_TRUNC, 0644);
3.  FileOutputStream\* output = new FileOutputStream(fd);
4.  CHECK(google::protobuf::TextFormat::Print(proto, output));
5.  delete output;
6.  close(fd);
7.  \}

**也就是说：先用ReadProtoFromBinaryFile函数将二进制文件读入proto里面，再将pro**to文件**读入txt文件就行了。**

**如果大家要建立一个工程的话，下面会有几个文件请放在一个文件夹中：**

#include <caffe/caffe.hpp>
        #include <google/protobuf/io/coded_stream.h>
        #include <google/protobuf/io/zero_copy_stream_impl.h>
        #include <google/protobuf/text_format.h>
        #include <algorithm>
        #include <iosfwd>
        #include <memory>
        #include <string>
        #include <utility>
        #include <vector>
        #include <iostream>
        #include "caffe/common.hpp"
        #include "caffe/proto/caffe.pb.h"
        #include "caffe/util/io.hpp"
    
        using namespace caffe;
        using namespace std;
        using google::protobuf::io::FileInputStream;
        using google::protobuf::io::FileOutputStream;
        using google::protobuf::io::ZeroCopyInputStream;
        using google::protobuf::io::CodedInputStream;
        using google::protobuf::io::ZeroCopyOutputStream;
        using google::protobuf::io::CodedOutputStream;
        using google::protobuf::Message;
    
        int main()
        {
            NetParameter proto;
        //    ReadProtoFromBinaryFile("../examples/mnist/lenet_iter_10000.caffemodel", &proto);  
           ReadProtoFromBinaryFile("/home/zy/zouyu/deeping/caffe/models/VGGNet/KITTI/SSD_300x300/KITTI_SSD_300x300_iter_80000.caffemodel", &proto);
           //把这个CAFFEMODEL通过这个函数读到proto中来。
           WriteProtoToTextFile(proto, "./test.txt");//把proto在上一个函数中读到的二进制流写到test.txt中。
            return 0;
        }
    ~                                                                                                                                                 
    ~                                                                                                                                                 
    ~                                                                                                                                                 
    ~                                                                                                                                                 
    ~

**其对应的CMakeLists.txt文件为：**

cmake_minimum_required (VERSION 2.8)
    
        project (pt_test)
    
        add_executable(pt_test pt.cpp)
    
        include_directories ( /home/zy/zouyu/deeping/caffe/include  
            /usr/local/include  
            /usr/local/cuda/include  
            /usr/include )
                   //请确保你的头文件与链接库存的文件路径正确。
        target_link_libraries(pt_test  
            /home/zy/zouyu/deeping/caffe/build/lib/libcaffe.so  
            /usr/lib/x86_64-linux-gnu/libglog.so  
            /usr/lib/x86_64-linux-gnu/libboost_system.so

**接下来就是编译的过程了：**

![20180607155224111][]

执行上面的cmake . && make 后会生成一个pt\_test可执行文件。当你执行过后，会生成test.txt。当你打开test.txt后，你会惊喜的发现，我原来训练的模型的相应参数都在这了。

在这个时候我的脑海中闪现一幅画面，看到了模型训练过程中图像从前端进去，经过各个卷积层、池化层、全连接层，在这一过程中利用递度下降等理论求解最小化的损失函数。最后网络停止更新，固化下来。生成了相应的MODEL，而整个训练集的DNA被以相应的形式记录了下来。如果后续要是有测试图片过来，就可以做相应的分类等工作了。

突然想到了《老子》中的一句话：“玄之又玄，众妙之门”。

相应的文件请点击如下链接：

[ look\_caffemodell][look_caffemodell]

[view plain]: https://blog.csdn.net/jiongnima/article/details/72904526#
[20180607155224111]: /images/20220523/0727bb9fb56e4f66aabccea1ca78a1d9.png
[look_caffemodell]: https://download.csdn.net/download/zouyu1746430162/10465111

发表评论取消回复

表情：

评论列表（有 0 条评论，205人围观）

还没有评论，来说两句吧...

相关阅读

相关庖丁解牛式读《Attention is all your need》

我的观点废话弄清楚Transformer模型内部的每一个细节尤为重要 attention机制首次被应用在nlp领域是在 2015年的一篇论文中：《Neural

àì夳堔傛蜴生んèń/ 2022年11月29日 12:52/ 0 赞/ 167 阅读

相关庖丁解牛-图解MySQL 8.0优化器查询转换篇

简介：在《庖丁解牛-图解MySQL 8.0优化器查询解析篇》一文中我们重点介绍了MySQL最新版本8.0.25关于SQL基本元素表、列、函数、聚合、分组、排序等元素的解析、设

傷城~/ 2022年09月12日 00:55/ 0 赞/ 172 阅读

相关庖丁解牛-图解MySQL 8.0优化器查询解析篇

简介： SQL优化器本质上是一种高度抽象化的数据接口的实现，经过该设计，客户可以使用更通用且易于理解的SQL语言，对数据进行操作和处理，而不需要关注和抽象自己的数据接口，极大地

Bertha 。/ 2022年09月11日 05:28/ 0 赞/ 151 阅读

相关庖丁解牛-----winpcap源码彻底解密

[庖丁解牛-----winpcap源码彻底解密(一)][-----winpcap] 本文系转载:原文出处为:[http://blog.csdn.net/smileston

旧城等待，/ 2022年08月19日 05:57/ 0 赞/ 604 阅读

相关 oracle的rowid和rdba庖丁解牛

author：skate time：2010-08-23 -------------------- Oracle 8以下ROWID组成（也叫受限Rowid）为：FF

柔光的暖阳◎/ 2022年07月12日 15:51/ 0 赞/ 220 阅读

相关 caffemodel之庖丁解牛

很多时候算法工程师把精力更多的聚焦于当下最流行的框架。一旦出现了一些各个式样的流行框架，便迫不急待的在自己的数据集或者公共的数据集上跑。然后生成模型来验证准确率与检测速率。如果

- 日理万妓/ 2022年05月23日 10:56/ 0 赞/ 206 阅读

相关庖丁解牛之Webpack打包后的Vue模块代码结构（未压缩版）

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ub

妖狐艹你老母/ 2022年04月05日 03:50/ 0 赞/ 156 阅读

相关 JVM庖丁解牛之GC Roots

在Java内存运行时的各个部分中，程序计数器、虚拟机栈、本地方法栈三个区域随线程生而生灭而灭，栈中的栈帧随方法的进入和退出而有条不紊的进行入栈和出栈操作。因此这几个区域的内存分

小鱼儿/ 2022年04月04日 15:14/ 0 赞/ 506 阅读

相关庖丁解牛！深入剖析React Native下一代架构重构

Facebook在2018年6月官方宣布了大规模重构React Native的计划及重构路线图。目的是为了让React Native更加轻量化、更适应混合开发，接近甚至达到原生

Myth丶恋晨/ 2022年02月22日 08:09/ 0 赞/ 291 阅读

相关庖丁解牛！深入剖析React Native下一代架构重构

Facebook在2018年6月官方宣布了大规模重构React Native的计划及重构路线图。目的是为了让React Native更加轻量化、更适应混合开发，接近甚至达到原生

女爷i/ 2022年02月22日 00:37/ 0 赞/ 351 阅读