warp.py 文件源码

python
阅读 25 收藏 0 点赞 0 评论 0

项目:inverse-compositional-STN 作者: ericlin79119 项目源码 文件源码
def mtrx2vec(opt,pMtrx):
    [row0,row1,row2] = torch.unbind(pMtrx,dim=1)
    [e00,e01,e02] = torch.unbind(row0,dim=1)
    [e10,e11,e12] = torch.unbind(row1,dim=1)
    [e20,e21,e22] = torch.unbind(row2,dim=1)
    if opt.warpType=="translation": p = torch.stack([e02,e12],dim=1)
    if opt.warpType=="similarity": p = torch.stack([e00-1,e10,e02,e12],dim=1)
    if opt.warpType=="affine": p = torch.stack([e00-1,e01,e02,e10,e11-1,e12],dim=1)
    if opt.warpType=="homography": p = torch.stack([e00-1,e01,e02,e10,e11-1,e12,e20,e21],dim=1)
    return p

# warp the image
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号