diff options
| author | junyanz <junyanz@berkeley.edu> | 2017-04-22 03:45:58 -0700 |
|---|---|---|
| committer | junyanz <junyanz@berkeley.edu> | 2017-04-22 03:45:58 -0700 |
| commit | 318936a415814e4b7b7affd3efd69c2aa1331074 (patch) | |
| tree | d7c2902b3680991a3f2da76c68a4df9254b38d36 /util/visualizer.py | |
| parent | 8f72c929868a0be6ff132614b0a161c3ad63d778 (diff) | |
update README & fix test-time issues in visualizer
Diffstat (limited to 'util/visualizer.py')
| -rw-r--r-- | util/visualizer.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/util/visualizer.py b/util/visualizer.py index bca74b2..9a6da2a 100644 --- a/util/visualizer.py +++ b/util/visualizer.py @@ -9,7 +9,8 @@ class Visualizer(): def __init__(self, opt): # self.opt = opt self.display_id = opt.display_id - self.use_html = not opt.no_html + self.use_html = opt.isTrain and not opt.no_html + self.win_size = opt.display_winsize self.name = opt.name if self.display_id > 0: import visdom @@ -18,7 +19,6 @@ class Visualizer(): if self.use_html: self.web_dir = os.path.join(opt.checkpoints_dir, opt.name, 'web') self.img_dir = os.path.join(self.web_dir, 'images') - self.win_size = opt.display_winsize print('create web directory %s...' % self.web_dir) util.mkdirs([self.web_dir, self.img_dir]) |