diff options
| author | junyanz <junyanz@berkeley.edu> | 2017-04-18 03:56:29 -0700 |
|---|---|---|
| committer | junyanz <junyanz@berkeley.edu> | 2017-04-18 03:56:29 -0700 |
| commit | efef2906b0a3f4fd265823ff4b4b99ccebeb6d05 (patch) | |
| tree | 0b041af61c33b2262b336598b9ff402595ed5f3b /util/visualizer.py | |
| parent | e17139daa9f3af07acee72dbf186d7eaf9ad089e (diff) | |
add unet and update README
Diffstat (limited to 'util/visualizer.py')
| -rw-r--r-- | util/visualizer.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/util/visualizer.py b/util/visualizer.py index 0b8578e..4daf506 100644 --- a/util/visualizer.py +++ b/util/visualizer.py @@ -27,7 +27,7 @@ class Visualizer(): def display_current_results(self, visuals, epoch): if self.display_id > 0: # show images in the browser idx = 0 - for label, image_numpy in visuals: + for label, image_numpy in visuals.items(): image_numpy = np.flipud(image_numpy) self.display.image(image_numpy, title=label, win=self.display_id + idx) |