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"
|
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
|
||||||
stderr = sys.stderr
|
stderr = sys.stderr
|
||||||
sys.stderr = open(os.devnull, "w")
|
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
|
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")
|
tf.get_logger().setLevel("ERROR")
|
||||||
warnings.filterwarnings("ignore")
|
warnings.filterwarnings("ignore")
|
||||||
from scipy.signal import find_peaks
|
from scipy.signal import find_peaks
|
||||||
|
|
|
@ -1,8 +1,7 @@
|
||||||
# ocrd includes opencv, numpy, shapely, click
|
# ocrd includes opencv, numpy, shapely, click
|
||||||
ocrd >= 2.23.3
|
ocrd >= 2.23.3
|
||||||
keras >= 2.3.1, < 2.4
|
|
||||||
scikit-learn >= 0.23.2
|
scikit-learn >= 0.23.2
|
||||||
tensorflow-gpu >= 1.15, < 2
|
tensorflow-gpu >= 2.4.0
|
||||||
imutils >= 0.5.3
|
imutils >= 0.5.3
|
||||||
matplotlib
|
matplotlib
|
||||||
setuptools >= 50
|
setuptools >= 50
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue