reintroduce TF stderr / log silencer

pull/48/head
Robert Sachunsky 2 years ago
parent 9ef8259677
commit cade5dda73

@ -3,15 +3,22 @@ import gc
import itertools import itertools
import math import math
import os import os
import sys
from pathlib import Path from pathlib import Path
from typing import Union, List, Any from typing import Union, List, Any
import cv2 import cv2
import numpy as np 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 import tensorflow as tf
from tensorflow.python.keras.saving.save import load_model
sys.stderr = stderr
from mpire import WorkerPool from mpire import WorkerPool
from mpire.utils import make_single_arguments from mpire.utils import make_single_arguments
from tensorflow.python.keras.saving.save import load_model
class SbbBinarizer: class SbbBinarizer:

Loading…
Cancel
Save