qurator-spk/sbb_binarization
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_binarization.git synced 2025-07-30 21:29:54 +02:00
Code Issues Releases Wiki Activity

Merge pull request #35 from cneud/upgrade_to_tf2_with_tf1.compat_session

Browse source 
fix tf2 with tf1.compat session
This commit is contained in:
vahidrezanezhad 2022-04-25 16:52:26 +02:00 • committed by GitHub
commit 0fe5232488
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

5
requirements.txt
View file

@ -1,7 +1,6 @@
numpy >= 1.17.0, < 1.19.0 numpy
setuptools >= 41 setuptools >= 41
opencv-python-headless opencv-python-headless
ocrd >= 2.22.3 ocrd >= 2.22.3
keras >= 2.3.1, < 2.4
h5py < 3 h5py < 3
tensorflow-gpu >= 1.15, < 1.16 tensorflow-gpu >= 2.4.0

7
sbb_binarize/sbb_binarize.py
View file

@ -14,10 +14,11 @@ import cv2
environ['TF_CPP_MIN_LOG_LEVEL'] = '3' environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
stderr = sys.stderr stderr = sys.stderr
sys.stderr = open(devnull, 'w') sys.stderr = open(devnull, 'w')
from keras.models import load_model
from keras.backend import tensorflow_backend
sys.stderr = stderr
import tensorflow as tf import tensorflow as tf
from tensorflow.keras.models import load_model
from tensorflow.python.keras import backend as tensorflow_backend
sys.stderr = stderr
import logging import logging