diff options
| author | junyanz <junyanz@berkeley.edu> | 2017-10-06 10:46:43 -0700 |
|---|---|---|
| committer | junyanz <junyanz@berkeley.edu> | 2017-10-06 10:46:43 -0700 |
| commit | 7800d516596f1a25986b458cddf8b8785bcc7df8 (patch) | |
| tree | 56d57350e7104393f939ec7cc2e07c96840aaa27 /util | |
| parent | e986144cee13a921fd3ad68d564f820e8f7dd3b0 (diff) | |
support nc=1, add new leaerning rate policy and new initialization
Diffstat (limited to 'util')
| -rw-r--r-- | util/util.py | 2 | ||||
| -rw-r--r-- | util/visualizer.py | 3 |
2 files changed, 3 insertions, 2 deletions
diff --git a/util/util.py b/util/util.py index 781239f..4de0a74 100644 --- a/util/util.py +++ b/util/util.py @@ -11,6 +11,8 @@ import collections # |imtype|: the desired type of the converted numpy array def tensor2im(image_tensor, imtype=np.uint8): image_numpy = image_tensor[0].cpu().float().numpy() + if image_numpy.shape[0] == 1: + image_numpy = np.tile(image_numpy, (3, 1, 1)) image_numpy = (np.transpose(image_numpy, (1, 2, 0)) + 1) / 2.0 * 255.0 return image_numpy.astype(imtype) diff --git a/util/visualizer.py b/util/visualizer.py index 3733525..02a36b7 100644 --- a/util/visualizer.py +++ b/util/visualizer.py @@ -4,7 +4,7 @@ import ntpath import time from . import util from . import html - +from pdb import set_trace as st class Visualizer(): def __init__(self, opt): # self.opt = opt @@ -66,7 +66,6 @@ class Visualizer(): else: idx = 1 for label, image_numpy in visuals.items(): - #image_numpy = np.flipud(image_numpy) self.vis.image(image_numpy.transpose([2,0,1]), opts=dict(title=label), win=self.display_id + idx) idx += 1 |