rm view

fef47dcc · WenmuZhou · d9f1fed7 · fef47dcc
Commit fef47dcc authored Mar 28, 2022 by WenmuZhou
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

ppocr/losses/det_pse_loss.py ppocr/losses/det_pse_loss.py +3 -3

No files found.
--- a/ppocr/losses/det_pse_loss.py
+++ b/ppocr/losses/det_pse_loss.py
@@ -121,9 +121,9 @@ class PSELoss(nn.Layer):

        if neg_num == 0:
            selected_mask = training_mask
-            selected_mask = selected_mask.view(
-                1, selected_mask.shape[0],
-                selected_mask.shape[1]).astype('float32')
+            selected_mask = selected_mask.reshape(
+                [1, selected_mask.shape[0], selected_mask.shape[1]]).astype(
+                    'float32')
            return selected_mask

        neg_score = paddle.masked_select(score, gt_text <= 0.5)