Merge pull request #74 from cneud/main

Adapt to use tf1.compat session in tf2

qurator/eynollah/eynollah.py

@@ -20,10 +20,10 @@ import numpy as np
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
 stderr = sys.stderr
 sys.stderr = open(os.devnull, "w")
-from keras import backend as K
-from keras.models import load_model
-sys.stderr = stderr
 import tensorflow as tf
+from tensorflow.python.keras import backend as K
+from tensorflow.keras.models import load_model
+sys.stderr = stderr
 tf.get_logger().setLevel("ERROR")
 warnings.filterwarnings("ignore")
 from scipy.signal import find_peaks

requirements.txt

@@ -1,8 +1,7 @@
 # ocrd includes opencv, numpy, shapely, click
 ocrd >= 2.23.3
-keras >= 2.3.1, < 2.4
 scikit-learn >= 0.23.2
-tensorflow-gpu >= 1.15, < 2
+tensorflow-gpu >= 2.4.0
 imutils >= 0.5.3
 matplotlib
 setuptools >= 50