Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/mmcv.git" did not exist on "a02a48157638fe6abf0ad4a3bd702ec91ca5414e"
Commit fef47dcc authored by WenmuZhou's avatar WenmuZhou
Browse files

rm view

parent d9f1fed7
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
ppocr/losses/det_pse_loss.py
View file @ fef47dcc
...@@ -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