mirror of
https://github.com/qurator-spk/eynollah.git
synced 2025-06-08 19:59:56 +02:00
Merge pull request #74 from cneud/main
Adapt to use tf1.compat session in tf2
This commit is contained in:
commit
ae1990a48e
2 changed files with 4 additions and 5 deletions
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue