Fixed bugs (#54)

* first commit test

* keras resnet test

* fixed bugs FPNs

* fixed bugs FPNs

* FPNs BatchNorm

* FPNs BatchNorm

* FPN

* fixed bugs

* fixed bugs

Author: mbroisinBI

keras_resnet/models/_feature_pyramid_2d.py

@@ -18,14 +18,14 @@
 
 class FPN2D(keras.Model):
     def __init__(
-        self,
-        inputs,
-        blocks,
-        block,
-        freeze_bn=True,
-        numerical_names=None,
-        *args,
-        **kwargs
+            self,
+            inputs,
+            blocks,
+            block,
+            freeze_bn=True,
+            numerical_names=None,
+            *args,
+            **kwargs
     ):
         if keras.backend.image_data_format() == "channels_last":
             axis = 3
@@ -74,14 +74,6 @@ def __init__(
             size=(2, 2)
         )(pyramid_5)
 
-        pyramid_5 = keras.layers.Conv2D(
-            filters=256,
-            kernel_size=3,
-            strides=1,
-            padding="same",
-            name="p5"
-        )(pyramid_5)
-
         pyramid_4 = keras.layers.Conv2D(
             filters=256,
             kernel_size=1,
@@ -154,13 +146,7 @@ def __init__(
             name="p2"
         )(pyramid_2)
 
-        pyramid_6 = keras.layers.Conv2D(
-            filters=256,
-            kernel_size=3,
-            strides=2,
-            padding="same",
-            name="p6"
-        )(c5)
+        pyramid_6 = keras.layers.MaxPooling2D(strides=2, name="p6")(pyramid_5)
 
         outputs = [
             pyramid_2,