qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-02-20 16:32:03 +01:00
Code Issues Projects Releases Wiki Activity

enhancer/mbreorder: use std Keras data loader for classification

Browse source
This commit is contained in:
Robert Sachunsky 2026-02-17 18:39:30 +01:00
parent 003c88f18a
commit a9496bbc70
2 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/eynollah/image_enhancer.py
View file

@ -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

4
src/eynollah/mb_ro_on_layout.py
View file

@ -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