Skip to content

GitLab

"tests/vscode:/vscode.git/clone" did not exist on "40aa47b998b0203f6aca0759df9f5eeefd64fcc7"
Unverified Commit 4348fc35 authored by Double_V's avatar Double_V Committed by GitHub
Browse files

Merge pull request #5803 from WenmuZhou/pse 

rm view
parents e55a72e0 fef47dcc
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
ppocr/losses/det_pse_loss.py
View file @ 4348fc35
...@@ -121,9 +121,9 @@ class PSELoss(nn.Layer): ...@@ -121,9 +121,9 @@ class PSELoss(nn.Layer):
if neg_num == 0: if neg_num == 0:
selected_mask = training_mask selected_mask = training_mask
selected_mask = selected_mask.view( selected_mask = selected_mask.reshape(
1, selected_mask.shape[0], [1, selected_mask.shape[0], selected_mask.shape[1]]).astype(
selected_mask.shape[1]).astype('float32') 'float32')
return selected_mask return selected_mask
neg_score = paddle.masked_select(score, gt_text <= 0.5) neg_score = paddle.masked_select(score, gt_text <= 0.5)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment