From 67fca82f384074028445f47fbcfdae44534668e1 Mon Sep 17 00:00:00 2001 From: Robert Sachunsky Date: Tue, 17 Feb 2026 18:06:08 +0100 Subject: [PATCH] fix missing import in 27f43c17 --- src/eynollah/training/models.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/src/eynollah/training/models.py b/src/eynollah/training/models.py index 4af4949..ba61764 100644 --- a/src/eynollah/training/models.py +++ b/src/eynollah/training/models.py @@ -441,7 +441,7 @@ def cnn_rnn_ocr_model(image_height=None, image_width=None, n_classes=None, max_s x = Conv2D(64,kernel_size=(3,3),padding="same")(x) x = BatchNormalization(name="bn2")(x) x = Activation("relu", name="relu2")(x) - x = MaxPool2D(pool_size=(1,2),strides=(1,2))(x) + x = MaxPooling2D(pool_size=(1,2),strides=(1,2))(x) x = Conv2D(128,kernel_size=(3,3),padding="same")(x) x = BatchNormalization(name="bn3")(x) @@ -449,7 +449,7 @@ def cnn_rnn_ocr_model(image_height=None, image_width=None, n_classes=None, max_s x = Conv2D(128,kernel_size=(3,3),padding="same")(x) x = BatchNormalization(name="bn4")(x) x = Activation("relu", name="relu4")(x) - x = MaxPool2D(pool_size=(1,2),strides=(1,2))(x) + x = MaxPooling2D(pool_size=(1,2),strides=(1,2))(x) x = Conv2D(256,kernel_size=(3,3),padding="same")(x) x = BatchNormalization(name="bn5")(x) @@ -457,7 +457,7 @@ def cnn_rnn_ocr_model(image_height=None, image_width=None, n_classes=None, max_s x = Conv2D(256,kernel_size=(3,3),padding="same")(x) x = BatchNormalization(name="bn6")(x) x = Activation("relu", name="relu6")(x) - x = MaxPool2D(pool_size=(2,2),strides=(2,2))(x) + x = MaxPooling2D(pool_size=(2,2),strides=(2,2))(x) x = Conv2D(image_width,kernel_size=(3,3),padding="same")(x) x = BatchNormalization(name="bn7")(x) @@ -465,8 +465,8 @@ def cnn_rnn_ocr_model(image_height=None, image_width=None, n_classes=None, max_s x = Conv2D(image_width,kernel_size=(16,1))(x) x = BatchNormalization(name="bn8")(x) x = Activation("relu", name="relu8")(x) - x2d = MaxPool2D(pool_size=(1,2),strides=(1,2))(x) - x4d = MaxPool2D(pool_size=(1,2),strides=(1,2))(x2d) + x2d = MaxPooling2D(pool_size=(1,2),strides=(1,2))(x) + x4d = MaxPooling2D(pool_size=(1,2),strides=(1,2))(x2d) new_shape = (x.shape[1]*x.shape[2], x.shape[3])