p83571920
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!22803 Icnet backbone num_class 

Merge pull request !22803 from TuDouNi/master
This commit is contained in:
i-robot 2021-09-06 08:52:52 +00:00 committed by Gitee
parent 5afb55d2e0 34cae0fcb8
commit 2936d4fe66
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/research/cv/ICNet/Res50V1_PRE/src/config.py
View File

@ -22,7 +22,7 @@ cfg = ed({
}) })
config1 = ed({ config1 = ed({
"class_num": 1000, "class_num": 1001,
"batch_size": 256, "batch_size": 256,
"loss_scale": 1024, "loss_scale": 1024,
"momentum": 0.9, "momentum": 0.9,

2
model_zoo/research/cv/ICNet/eval.py
View File

@ -88,7 +88,7 @@ class Evaluator:
self.metric.update(output, mask) self.metric.update(output, mask)
list_time.append(step_time) list_time.append(step_time)
pixAcc, mIoU = self.metric.get() mIoU, pixAcc = self.metric.get()
average_time = sum(list_time) / len(list_time) average_time = sum(list_time) / len(list_time)