mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

update etsnet to psenet

This commit is contained in:
anzhengqi 2021-09-06 16:15:51 +08:00
parent c759dea52f
commit 3e821e6483
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/official/cv/psenet/src/PSENET/psenet.py
View File

@ -26,7 +26,7 @@ from .fpn import FPN
class PSENet(nn.Cell): class PSENet(nn.Cell):
def __init__(self, config): def __init__(self, config):
super(PESNet, self).__init__() super(PSENet, self).__init__()
self.kernel_num = config.KERNEL_NUM self.kernel_num = config.KERNEL_NUM
self.inference = config.INFERENCE self.inference = config.INFERENCE
if config.INFERENCE: if config.INFERENCE: