mirror of
https://github.com/qurator-spk/sbb_binarization.git
synced 2025-06-09 04:10:00 +02:00
binarizer: reset TF session prior to predict()
This commit is contained in:
parent
02423648ec
commit
ad50ed287d
1 changed files with 4 additions and 0 deletions
|
@ -15,6 +15,7 @@ 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
|
||||
|
||||
|
@ -42,8 +43,10 @@ class SbbBinarizer:
|
|||
config.gpu_options.allow_growth = True
|
||||
|
||||
self.session = tf.Session(config=config) # tf.InteractiveSession()
|
||||
tensorflow_backend.set_session(self.session)
|
||||
|
||||
def end_session(self):
|
||||
tensorflow_backend.clear_session()
|
||||
self.session.close()
|
||||
del self.session
|
||||
|
||||
|
@ -55,6 +58,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)
|
||||
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