mirror of
https://github.com/qurator-spk/sbb_binarization.git
synced 2025-06-09 04:10:00 +02:00
use tensorflow Keras backend
This commit is contained in:
parent
1a4207c614
commit
b7aeb1ae8b
1 changed files with 7 additions and 6 deletions
|
@ -14,10 +14,11 @@ import cv2
|
|||
environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
|
||||
stderr = sys.stderr
|
||||
sys.stderr = open(devnull, 'w')
|
||||
from keras.models import load_model
|
||||
from keras.backend import tensorflow_backend
|
||||
sys.stderr = stderr
|
||||
import tensorflow as tf
|
||||
from tensorflow.keras.models import load_model, Model
|
||||
from tensorflow.python.keras import backend as K
|
||||
sys.stderr = stderr
|
||||
|
||||
|
||||
import logging
|
||||
|
||||
|
@ -43,10 +44,10 @@ class SbbBinarizer:
|
|||
config.gpu_options.allow_growth = True
|
||||
|
||||
self.session = tf.compat.v1.Session(config=config) # tf.InteractiveSession()
|
||||
tensorflow_backend.set_session(self.session)
|
||||
K.set_session(self.session)
|
||||
|
||||
def end_session(self):
|
||||
tensorflow_backend.clear_session()
|
||||
K.clear_session()
|
||||
self.session.close()
|
||||
del self.session
|
||||
|
||||
|
@ -58,7 +59,7 @@ class SbbBinarizer:
|
|||
return model, model_height, model_width, n_classes
|
||||
|
||||
def predict(self, model_in, img, use_patches):
|
||||
tensorflow_backend.set_session(self.session)
|
||||
K.set_session(self.session)
|
||||
model, model_height, model_width, n_classes = model_in
|
||||
|
||||
img_org_h = img.shape[0]
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue