Make model cheaper to run

Author: ankandrew

fast_plate_ocr/model/models.py

@@ -33,11 +33,10 @@ def modelo_2m(
     input_tensor = Input((h, w, 1))
     x = Rescaling(1.0 / 255)(input_tensor)
     # Backbone
-    x, _ = block_bn(x)
-    x, _ = block_bn(x, k=3, n_c=32, s=1, padding="same")
-    x, _ = block_bn(x, k=3, n_c=32, s=1, padding="same")
+    x, _ = block_bn(x, k=3, n_c=32, s=2, padding="same")
+    x, _ = block_bn(x, k=3, n_c=64, s=1, padding="same")
     x, _ = block_bn(x, k=1, n_c=64, s=1, padding="same")
-    x = MaxPool2D(pool_size=(3, 3), strides=(3, 3), padding="same")(x)
+    x = MaxPool2D(pool_size=(2, 2), strides=(2, 2), padding="same")(x)
     x, _ = block_bn(x, k=3, n_c=64, s=1, padding="same")
     x, _ = block_bn(x, k=3, n_c=128, s=1, padding="same")
     x, _ = block_bn(x, k=1, n_c=128, s=1, padding="same")