基於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.

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章