IcNet eval time

2021-08-25 18:36:40 +08:00 · 2021-08-25 18:36:40 +08:00 · a24aea5233
parent 3ad641cb65
commit a24aea5233
1 changed files with 2 additions and 2 deletions
--- a/model_zoo/research/cv/ICNet/eval.py
+++ b/model_zoo/research/cv/ICNet/eval.py
@ -83,8 +83,8 @@ class Evaluator:
            end_time = time.time()
            step_time = end_time - start_time

-            output = np.array(output)
-            mask = np.expand_dims(mask, axis=0)
+            output = output.asnumpy()
+            mask = np.expand_dims(mask.asnumpy(), axis=0)
            self.metric.update(output, mask)
            list_time.append(step_time)