From 193071500c5ee48e6016963ba478b181883d37ba Mon Sep 17 00:00:00 2001
From: Bubbliiiing <47347516+bubbliiiing@users.noreply.github.com>
Date: Thu, 4 Feb 2021 19:15:37 +0800
Subject: [PATCH] Update yolo4.py

---
 nets/yolo4.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/nets/yolo4.py b/nets/yolo4.py
index a3470d8..39d117f 100644
--- a/nets/yolo4.py
+++ b/nets/yolo4.py
@@ -103,11 +103,11 @@ def yolo_body(inputs, num_anchors, num_classes):
     #   y3=(batch_size,52,52,3,85)
     #---------------------------------------------------#
     P3_output = DarknetConv2D_BN_Leaky(256, (3,3))(P3)
-    P3_output = DarknetConv2D(num_anchors*(num_classes+5), (1,1))(P3_output)
+    P3_output = DarknetConv2D(num_anchors*(num_classes+5), (1,1), kernel_initializer=keras.initializers.normal(mean=0.0, stddev=0.01))(P3_output)
 
     # 52,52,128 -> 26,26,256
     P3_downsample = ZeroPadding2D(((1,0),(1,0)))(P3)
-    P3_downsample = DarknetConv2D_BN_Leaky(256, (3,3), strides=(2,2), kernel_initializer=keras.initializers.normal(mean=0.0, stddev=0.01))(P3_downsample)
+    P3_downsample = DarknetConv2D_BN_Leaky(256, (3,3), strides=(2,2))(P3_downsample)
     # 26,26,256 + 26,26,256 -> 26,26,512
     P4 = Concatenate()([P3_downsample, P4])
     # 26,26,512 -> 26,26,256 -> 26,26,512 -> 26,26,256 -> 26,26,512 -> 26,26,256
-- 
GitLab