Skip to content
Snippets Groups Projects
Unverified Commit 16c458d8 authored by i-robot's avatar i-robot Committed by Gitee
Browse files

!3969 modify psenet scripts

Merge pull request !3969 from anzhengqi/modify-networks
parents 2ac98f51 63274043
No related branches found
No related tags found
No related merge requests found
......@@ -87,7 +87,7 @@ class DiceLoss(Cell):
neg_num = self.minimum(3 * pos_num, neg_num)
neg_num = self.cast(neg_num, mstype.int32)
neg_num = self.add(neg_num, self.negative_one_int32)
neg_num = neg_num + self.k - 1
neg_mask = self.less_equal(gt_text, self.threshold0)
ignore_score = self.fill(mstype.float32, (640, 640), -1e3)
neg_score = self.select(neg_mask, score, ignore_score)
......
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