Merge pull request #1951 from light1003/patch-1

Update det_basic_loss.py
2021-02-19 15:28:50 +08:00 · 2021-02-19 15:28:50 +08:00 · 85d340bd27
parent 1e5c6e43f3 4a8fc9176d
commit 85d340bd27
1 changed files with 1 additions and 1 deletions
--- a/ppocr/losses/det_basic_loss.py
+++ b/ppocr/losses/det_basic_loss.py
@ -200,6 +200,6 @@ def ohem_batch(scores, gt_texts, training_masks, ohem_ratio):
                i, :, :], ohem_ratio))

    selected_masks = np.concatenate(selected_masks, 0)
-    selected_masks = paddle.to_variable(selected_masks)
+    selected_masks = paddle.to_tensor(selected_masks)

    return selected_masks