multiple preview file support for colab

2019-04-22 00:49:50 +04:00
parent f54cc744bd
commit 44aa530fda
1 changed files with 4 additions and 1 deletions
--- 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))) ) ]