From 962a628d17848fe5c46d298bacba21d237a8b9e4 Mon Sep 17 00:00:00 2001 From: Double_V Date: Wed, 9 Jun 2021 19:30:52 +0800 Subject: [PATCH] remove tmp variable --- ppocr/modeling/transforms/tps.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/ppocr/modeling/transforms/tps.py b/ppocr/modeling/transforms/tps.py index 13220991..dcce6246 100644 --- a/ppocr/modeling/transforms/tps.py +++ b/ppocr/modeling/transforms/tps.py @@ -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 [ -- GitLab