diff options
Diffstat (limited to 'util/util.py')
| -rw-r--r-- | util/util.py | 19 |
1 files changed, 3 insertions, 16 deletions
diff --git a/util/util.py b/util/util.py index 4de0a74..26b259a 100644 --- a/util/util.py +++ b/util/util.py @@ -2,11 +2,13 @@ from __future__ import print_function import torch import numpy as np from PIL import Image -import inspect, re +import inspect +import re import numpy as np import os import collections + # Converts a Tensor into a Numpy array # |imtype|: the desired type of the converted numpy array def tensor2im(image_tensor, imtype=np.uint8): @@ -34,21 +36,6 @@ def save_image(image_numpy, image_path): image_pil = Image.fromarray(image_numpy) image_pil.save(image_path) -def info(object, spacing=10, collapse=1): - """Print methods and doc strings. - Takes module, class, list, dictionary, or string.""" - methodList = [e for e in dir(object) if isinstance(getattr(object, e), collections.Callable)] - processFunc = collapse and (lambda s: " ".join(s.split())) or (lambda s: s) - print( "\n".join(["%s %s" % - (method.ljust(spacing), - processFunc(str(getattr(object, method).__doc__))) - for method in methodList]) ) - -def varname(p): - for line in inspect.getframeinfo(inspect.currentframe().f_back)[3]: - m = re.search(r'\bvarname\s*\(\s*([A-Za-z_][A-Za-z0-9_]*)\s*\)', line) - if m: - return m.group(1) def print_numpy(x, val=True, shp=False): x = x.astype(np.float64) |