diff --git a/sbb_binarize/sbb_binarize.py b/sbb_binarize/sbb_binarize.py index 70a81cf..bb58f36 100644 --- a/sbb_binarize/sbb_binarize.py +++ b/sbb_binarize/sbb_binarize.py @@ -2,19 +2,20 @@ Tool to load model and binarize a given image. """ -from os import listdir +import sys +from os import listdir, environ, devnull from os.path import join from warnings import catch_warnings, simplefilter import numpy as np import cv2 +environ['TF_CPP_MIN_LOG_LEVEL'] = '3' +stderr = sys.stderr +sys.stderr = open(devnull, 'w') from keras.models import load_model +sys.stderr = stderr import tensorflow as tf -# XXX better to set env var before tensorflow import to suppress those specific warnings -with catch_warnings(): - simplefilter("ignore") - def resize_image(img_in, input_height, input_width): return cv2.resize(img_in, (input_width, input_height), interpolation=cv2.INTER_NEAREST)