From 00336b7df5cba4bfaef4cdc7f905ad17a36573be Mon Sep 17 00:00:00 2001
From: wsq3 <877518222@qq.com>
Date: Fri, 23 Apr 2021 09:22:50 +0800
Subject: [PATCH] optimized the network of yolov3_darknet53

---
 model_zoo/official/cv/yolov3_darknet53/eval.py       | 2 +-
 model_zoo/official/cv/yolov3_darknet53/src/config.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/model_zoo/official/cv/yolov3_darknet53/eval.py b/model_zoo/official/cv/yolov3_darknet53/eval.py
index 7e230e68cf..afa4811f0b 100644
--- a/model_zoo/official/cv/yolov3_darknet53/eval.py
+++ b/model_zoo/official/cv/yolov3_darknet53/eval.py
@@ -173,7 +173,7 @@ class DetectionEngine:
                 y = y.reshape(-1)
                 w = w.reshape(-1)
                 h = h.reshape(-1)
-                cls_emb = cls_emb.reshape(-1, 80)
+                cls_emb = cls_emb.reshape(-1, self.num_classes)
                 conf = conf.reshape(-1)
                 cls_argmax = cls_argmax.reshape(-1)
 
diff --git a/model_zoo/official/cv/yolov3_darknet53/src/config.py b/model_zoo/official/cv/yolov3_darknet53/src/config.py
index 16831b048d..8cf6d18d99 100644
--- a/model_zoo/official/cv/yolov3_darknet53/src/config.py
+++ b/model_zoo/official/cv/yolov3_darknet53/src/config.py
@@ -62,7 +62,7 @@ class ConfigYOLOV3DarkNet53:
                      (116, 90),
                      (156, 198),
                      (373, 326)]
-    out_channel = 255
+    out_channel = 3 * (num_classes + 5)
 
     # test_param
     test_img_shape = [416, 416]