基于CAFFE的多任务训练

背景

两任务分别为：openpose人体姿态估计任务以及yolov3检测任务。

框架：CAFFE

解决方案

1、数据层

法一：可更改DATA层，输出两个lbael，分别对应于人体姿态估计任务以及检测任务。结构如下图所示。

法二：另一种方法简单一些，caffe支持两个data层，其结构如下图所示。每层数据格式为N*C*H*W，N表示数据个数，C表示通道数，H与W分别表示feature map的宽与高。首先将两个data层，按照第一维进行合并，N=N1+N2，N1与N2分别表示DATA1的batch与DATA2的batch的大小。经过公共网络后，将第一维N拆解为N1与N2.