From a4c0cf0a475dbe8934fab3ae699512c435029e2d Mon Sep 17 00:00:00 2001 From: Konstantin Baierer Date: Thu, 15 Oct 2020 17:26:13 +0200 Subject: [PATCH] shut up keras/tensorflow --- sbb_binarize/sbb_binarize.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) 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)