Deep Image Homography Estimation理解

文獻:Deep Image Homography Estimation,下載地址

輸入:128x128x2

Padding:'SAME'

池化步長:2

迴歸模型(HomographyNet-Regression):

conv1 3x3 : 128x128x64

conv2 3x3 : 128x128x64

maxpooling1 2x2: 64x64x64

 

conv3 3x3 : 64x64x64

conv4 3x3 : 64x64x64

maxpooling2 2x2: 32x32x64

 

conv5 3x3 : 32x32x128

conv6 3x3 : 32x32x128

maxpooling3 2x2: 16x16x128

 

conv7 3x3 : 16x16x128

conv8 3x3 : 16x16x128

 

fully connect1: 1024x1

fully connect2: 8x1

 

loss function:

 \frac{1}{2}*\left \| p\left ( x \right ) -q\left ( x \right )\right \|^2

 

分類模型(HomographyNet-Classification):

conv1 3x3 : 128x128x64

conv2 3x3 : 128x128x64

maxpooling1 2x2: 64x64x64

 

conv3 3x3 : 64x64x64

conv4 3x3 : 64x64x64

maxpooling2 2x2: 32x32x64

 

conv5 3x3 : 32x32x128

conv6 3x3 : 32x32x128

maxpooling3 2x2: 16x16x128

 

conv7 3x3 : 16x16x128

conv8 3x3 : 16x16x128

 

fully connect1: 1024x1

fully connect2: 8x21

softmax

 

loss function:

-\sum p\left ( x \right )log\left ( q\left ( x \right ) \right )

 

訓練方式:SGD(隨機梯度下降法) ,momentum =  0.9

訓練數據製作:

https://img-blog.csdnimg.cn/20181225205129538.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3pjZzE5NDI=,size_16,color_FFFFFF,t_70

訓練標籤:

\left [ \Delta u1,\Delta v1,\Delta u2,\Delta v2,\Delta u3,\Delta v3,\Delta u4,\Delta v4 \right ],與放射矩陣H一一對應

訓練設置:conv8與fully connect1需要添加dropout=0.5

測試數據集:MS-COCO

運行效率:NVIDIA Titan X GPU, 300fps

 

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