DRML（2016-CVPR）重现过程记录---（1）caffe+DRML源码编译

来源：互联网发布：写日志软件编辑：程序博客网时间：2024/06/12 23:16

参考资料和说明

1，DRML论文：http://www.cv-foundation.org/openaccess/content_cvpr_2016/papers/Zhao_Deep_Region_and_CVPR_2016_paper.pdf

Zhao K, Chu W S, Zhang H. Deep Region and Multi-label Learning for Facial Action Unit Detection[C]// IEEE Conference on Computer Vision and Pattern Recognition. IEEE, 2016:3391-3399.

作者定义了自己的新的层结构做区域学习多标签检测，来提升AU检测的效果。文章对12类AU的检测进行了训练和测试，分别是AU1,AU12,AU15,AU17,AU2，AU20,AU25,AU26,AU4,AU5,AU6,AU9

2，作者提供了部分源码：https://github.com/zkl20061823/DRML

该源码包含作者新定义的网络层结构和prototxt

重现文章的思路：在现有caffe基础上添加作者自定义的层

本文讲解caffe+DRML源码编译，后续处理我会在后面的博文中讲到。

注意：作者的DRML源码是在caffe-master_20150520_1版本（有点像caffe rc2版本，但是不是完全的caffe rc2，所以我邮件找作者问了，作者提供了他们当时用的Caffe版本）基础上做的，但是新版本中很多函数和变量已经变了，我现在用新的caffe版本+DRML源码编译，所以需要知道DRML在原来caffe版本上有什么改变，以及如何在caffe中添加新的层定义。

感谢作者提供了他们当时用的caffe源码！

作者使用的caffe版本与DRML源码对比（beyondcompare结果）

（注：caffe.proto是我自己修改的，作者没有提供。作者提供了common_layers.hpp,data_layers.hpp,loss_layers.hpp以及box_layer.cpp,box_layer.cu,image_data_layer_multilabel.cpp,multi_sigmoid_cross_entropy_loss_layer.cpp,

multi_sigmoid_cross_entropy_loss_layer.cu,splice.cpp,cplice.cu）