github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 04:52:13 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

multiple preview file support for colab

Browse source
This commit is contained in:
iperov 2019-04-22 00:49:50 +04:00
parent f54cc744bd
commit 44aa530fda
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
models/ModelBase.py
View file

@ -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))) ) ]