From 070e93856888a5f57a3dcf4c588960cd947208c5 Mon Sep 17 00:00:00 2001 From: iperov Date: Tue, 26 Mar 2019 18:11:33 +0400 Subject: [PATCH] colab: final fix for preview filepath --- models/ModelBase.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/models/ModelBase.py b/models/ModelBase.py index 4581a7e..e487122 100644 --- a/models/ModelBase.py +++ b/models/ModelBase.py @@ -367,16 +367,17 @@ class ModelBase(object): if self.iter % 10 == 0: plist = [] + if io.is_colab(): - plist += [ (self.get_previews()[0][1], '_preview.jpg') ] + plist += [ (self.get_previews()[0][1], self.get_strpath_storage_for_file('preview.jpg') ) ] if self.write_preview_history: - plist += [ (self.get_static_preview(), '%.6d.jpg' %(self.iter) ) ] + plist += [ (self.get_static_preview(), str (self.preview_history_path / ('%.6d.jpg' % (self.iter))) ) ] - for preview, filename in plist: + for preview, filepath in plist: preview_lh = ModelBase.get_loss_history_preview(self.loss_history, self.iter, preview.shape[1], preview.shape[2]) img = (np.concatenate ( [preview_lh, preview], axis=0 ) * 255).astype(np.uint8) - cv2_imwrite ( str (self.preview_history_path / filename), img ) + cv2_imwrite (filepath, img ) self.iter += 1