From 134451267f13abf8c35d089ec8ac7e2c531e7da5 Mon Sep 17 00:00:00 2001 From: Baek JeongHun Date: Wed, 22 Jan 2020 03:08:24 +0900 Subject: [PATCH] Update transformation.py --- modules/transformation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/transformation.py b/modules/transformation.py index 311838a..875d1ae 100755 --- a/modules/transformation.py +++ b/modules/transformation.py @@ -32,7 +32,7 @@ class TPS_SpatialTransformerNetwork(nn.Module): build_P_prime_reshape = build_P_prime.reshape([build_P_prime.size(0), self.I_r_size[0], self.I_r_size[1], 2]) if torch.__version__ > "1.2.0": - batch_I_r = F.grid_sample(batch_I, build_P_prime_reshape, padding_mode='border', align_corner=True) + batch_I_r = F.grid_sample(batch_I, build_P_prime_reshape, padding_mode='border', align_corners=True) else: batch_I_r = F.grid_sample(batch_I, build_P_prime_reshape, padding_mode='border')