From a9496bbc7079d11706e34d1fcef4a0269fe23117 Mon Sep 17 00:00:00 2001 From: Robert Sachunsky Date: Tue, 17 Feb 2026 18:39:30 +0100 Subject: [PATCH] enhancer/mbreorder: use std Keras data loader for classification --- src/eynollah/image_enhancer.py | 6 ++++-- src/eynollah/mb_ro_on_layout.py | 4 +++- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/src/eynollah/image_enhancer.py b/src/eynollah/image_enhancer.py index babbd55..67145a3 100644 --- a/src/eynollah/image_enhancer.py +++ b/src/eynollah/image_enhancer.py @@ -15,11 +15,13 @@ from pathlib import Path import gc import cv2 -from keras.models import Model import numpy as np -import tensorflow as tf # type: ignore from skimage.morphology import skeletonize +os.environ['TF_USE_LEGACY_KERAS'] = '1' # avoid Keras 3 after TF 2.15 +import tensorflow as tf # type: ignore +from tensorflow.keras.models import Model + from .model_zoo import EynollahModelZoo from .utils.resize import resize_image from .utils.pil_cv2 import pil2cv diff --git a/src/eynollah/mb_ro_on_layout.py b/src/eynollah/mb_ro_on_layout.py index eec544c..22fe97b 100644 --- a/src/eynollah/mb_ro_on_layout.py +++ b/src/eynollah/mb_ro_on_layout.py @@ -14,10 +14,12 @@ from pathlib import Path import xml.etree.ElementTree as ET import cv2 -from keras.models import Model import numpy as np import statistics + +os.environ['TF_USE_LEGACY_KERAS'] = '1' # avoid Keras 3 after TF 2.15 import tensorflow as tf +from tensorflow.keras.models import Model from .model_zoo import EynollahModelZoo from .utils.resize import resize_image