Update yolo4.py

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