From 2acf270b6e46cd01cf4f5dd0b884f9797778cedc Mon Sep 17 00:00:00 2001
From: dengkaipeng <dengkaipeng@baidu.com>
Date: Fri, 12 Apr 2019 05:57:23 +0000
Subject: [PATCH] fix yolov3 class_num.

---
 PaddleCV/yolov3/models/yolov3.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/PaddleCV/yolov3/models/yolov3.py b/PaddleCV/yolov3/models/yolov3.py
index 9ea4ef44..55c0667b 100644
--- a/PaddleCV/yolov3/models/yolov3.py
+++ b/PaddleCV/yolov3/models/yolov3.py
@@ -103,9 +103,12 @@ class YOLOv3(object):
             route, tip = yolo_detection_block(block, channel=512//(2**i), 
                                         is_test=(not self.is_train),
                                         name="yolo_block.{}".format(i))
+
+            # out channel number = mask_num * (5 + class_num)
+            num_filters = len(cfg.anchor_masks[i]) * (cfg.class_num + 5)
             block_out = fluid.layers.conv2d(
                 input=tip,
-                num_filters=255,
+                num_filters=num_filters,
                 filter_size=1,
                 stride=1,
                 padding=0,
-- 
GitLab