forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #1042 from dyning/develop 

add anno for tps
This commit is contained in:
dyning 2020-10-28 15:58:20 +08:00 committed by GitHub
parent 9f3b617ec7 4713f24dc1
commit 89ccf50bac
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
ppocr/modeling/stns/tps.py Executable file → Normal file
View File

@ -70,6 +70,13 @@ class LocalizationNetwork(object):
return initial_bias return initial_bias
def __call__(self, image): def __call__(self, image):
"""
Estimating parameters of geometric transformation
Args:
image: input
Return:
batch_C_prime: the matrix of the geometric transformation
"""
F = self.F F = self.F
loc_lr = self.loc_lr loc_lr = self.loc_lr
if self.model_name == "large": if self.model_name == "large":
@ -215,6 +222,14 @@ class GridGenerator(object):
return batch_C_ex_part_tensor return batch_C_ex_part_tensor
def __call__(self, batch_C_prime, I_r_size): def __call__(self, batch_C_prime, I_r_size):
"""
Generate the grid for the grid_sampler.
Args:
batch_C_prime: the matrix of the geometric transformation
I_r_size: the shape of the input image
Return:
batch_P_prime: the grid for the grid_sampler
"""
C = self.build_C() C = self.build_C()
P = self.build_P(I_r_size) P = self.build_P(I_r_size)
inv_delta_C = self.build_inv_delta_C(C).astype('float32') inv_delta_C = self.build_inv_delta_C(C).astype('float32')