OrgLion-ML
/
deep-text-recognition-benchmark
mirror of https://github.com/clovaai/deep-text-recognition-benchmark.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Update transformation.py

This commit is contained in:
Baek JeongHun 2020-01-22 03:08:24 +09:00 committed by GitHub
parent 78d7ed96a9
commit 134451267f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/transformation.py
View File

@ -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]) 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": 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: else:
batch_I_r = F.grid_sample(batch_I, build_P_prime_reshape, padding_mode='border') batch_I_r = F.grid_sample(batch_I, build_P_prime_reshape, padding_mode='border')