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
评论列表
文章目录