From 44aa530fdab9d5cc237a755a9a2c50fdabd831c8 Mon Sep 17 00:00:00 2001 From: iperov Date: Mon, 22 Apr 2019 00:49:50 +0400 Subject: [PATCH] multiple preview file support for colab --- models/ModelBase.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/models/ModelBase.py b/models/ModelBase.py index 83dbdbe..ea59f7d 100644 --- a/models/ModelBase.py +++ b/models/ModelBase.py @@ -373,7 +373,10 @@ class ModelBase(object): plist = [] if io.is_colab(): - plist += [ (self.get_previews()[0][1], self.get_strpath_storage_for_file('preview.jpg') ) ] + previews = self.get_previews() + for i in range(previews): + name, bgr = previews[i] + plist += [ (bgr, self.get_strpath_storage_for_file('preview_%s.jpg' % (name) ) ) ] if self.write_preview_history: plist += [ (self.get_static_preview(), str (self.preview_history_path / ('%.6d.jpg' % (self.iter))) ) ]