opensource/PaddleOCR
3
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2026-09-24 23:33:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove tmp variable

Browse Source
This commit is contained in:
Double_V
2021-06-09 19:30:52 +08:00
committed by GitHub
parent 616ad6a179
commit 962a628d17
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ppocr/modeling/transforms/tps.py
+1 -3
View File
@@ -231,9 +231,7 @@ class GridGenerator(nn.Layer):
""" Return inv_delta_C which is needed to calculate T """
F = self.F
hat_eye = paddle.eye(F, dtype='float64') # F x F
tmp1 = C.reshape([1, F, 2])
tmp2 = C.reshape([F, 1, 2])
hat_C = paddle.norm(tmp1 - tmp2, axis=2) + hat_eye
hat_C = paddle.norm(C.reshape([1, F, 2]) - C.reshape([F, 1, 2]), axis=2) + hat_eye
hat_C = (hat_C**2) * paddle.log(hat_C)
delta_C = paddle.concat( # F+3 x F+3
[