diff options
| author | junyanz <junyanz@berkeley.edu> | 2018-01-14 17:13:18 -0800 |
|---|---|---|
| committer | junyanz <junyanz@berkeley.edu> | 2018-01-14 17:13:18 -0800 |
| commit | 77385c1e26ed5e5c13b32c9582de3e9302a45e6e (patch) | |
| tree | f2d1d4cbc8dd9f62f323129d1cdb955d735f36d7 /util | |
| parent | a956f8ca3521a23071d2d260bdcd59ccfb0772a1 (diff) | |
update README
Diffstat (limited to 'util')
| -rw-r--r-- | util/util.py | 19 | ||||
| -rw-r--r-- | util/visualizer.py | 1 |
2 files changed, 3 insertions, 17 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) diff --git a/util/visualizer.py b/util/visualizer.py index 58d1a2a..8bec8df 100644 --- a/util/visualizer.py +++ b/util/visualizer.py @@ -5,7 +5,6 @@ import time from . import util from . import html from scipy.misc import imresize -from pdb import set_trace as st class Visualizer(): |
