reintroduce TF stderr / log silencer

This commit is contained in:
Robert Sachunsky 2023-02-14 23:09:40 +00:00
parent 9ef8259677
commit cade5dda73

View file

@ -3,15 +3,22 @@ import gc
import itertools
import math
import os
import sys
from pathlib import Path
from typing import Union, List, Any
import cv2
import numpy as np
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
stderr = sys.stderr
sys.stderr = open(os.devnull, 'w')
import tensorflow as tf
from tensorflow.python.keras.saving.save import load_model
sys.stderr = stderr
from mpire import WorkerPool
from mpire.utils import make_single_arguments
from tensorflow.python.keras.saving.save import load_model
class SbbBinarizer: