imwarp matlab,matlab、opencv、pytorch实现仿射变换

matlab、opencv、pytorch实现仿射变换

matlab、opencv、pytorch实现仿射变换

matlab

tform = affine2d(tform');

dst=imwarp(img,tform);

figure;imshow(dst);%会自动移到中心

opencv

需要自己平移

Mat Mt= (Mat_(2, 3) << 1,0,0,0,1,0);

warpAffine(srcimg, img, Mt,Size(N, M));//输出大小N*M

pytorch

和直觉相反，仿射变换矩阵应为A的逆。需要缩放平移才能显示完全

class AffineGridGen(Module):

def __init__(self, out_h=240, out_w=240, out_ch=3):

super(AffineGridGen, self).__init__()

self.out_h = out_h

self.out_w = out_w

self.out_ch = out_ch

def forward(self, theta):

theta = theta.contiguous()

batch_size = theta.size()[0]

out_size = torch.Size((batch_size, self.out_ch, self.out_h, self.out_w))

return F.affine_grid(theta, out_size)#仿射变换

img=cv2.imread('img.jpg')

img=transforms.ToTensor()(img).unsqueeze(0)

theta_batch=torch.Tensor(np.expand_dims(np.array([[1, 0, 0], [0, 1, 0]]), 0).astype(np.float32))

theta_batch = theta_batch.expand(1, 2, 3)

gridGen=AffineGridGen(500, 500)#输出大小 resize

sampling_grid=gridGen(theta_batch)

sampling_grid.data = sampling_grid.data *padding_factor * crop_factor

warped_image_batch = F.grid_sample(img,sampling_grid)

transforms.ToPILImage()(warped_image_batch[0]).show()

matlab、opencv、pytorch实现仿射变换相关教程