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://i-blog.csdnimg.cn/blog_migrate/4ea5b33ce2a81e5fc71d79f3d62df2bb.png

训练标签:

\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

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值