!22803 Icnet backbone num_class

Merge pull request !22803 from TuDouNi/master
2021-09-06 08:52:52 +00:00 · 2021-09-06 08:52:52 +00:00 · 2936d4fe66
parent 5afb55d2e0 34cae0fcb8
commit 2936d4fe66
2 changed files with 2 additions and 2 deletions
--- a/model_zoo/research/cv/ICNet/Res50V1_PRE/src/config.py
+++ b/model_zoo/research/cv/ICNet/Res50V1_PRE/src/config.py
@ -22,7 +22,7 @@ cfg = ed({
    })

 config1 = ed({
-    "class_num": 1000,
+    "class_num": 1001,
    "batch_size": 256,
    "loss_scale": 1024,
    "momentum": 0.9,
--- a/model_zoo/research/cv/ICNet/eval.py
+++ b/model_zoo/research/cv/ICNet/eval.py
@ -88,7 +88,7 @@ class Evaluator:
            self.metric.update(output, mask)
            list_time.append(step_time)

-        pixAcc, mIoU = self.metric.get()
+        mIoU, pixAcc = self.metric.get()

        average_time = sum(list_time) / len(list_time)