david/DeepFaceLab_Linux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix converter

Browse Source
This commit is contained in:
iperov
2019-08-26 11:03:57 +04:00
parent 78ddd51ffc
commit fb4837b3cb
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mainscripts/Converter.py
+7 -1
View File
@@ -210,10 +210,16 @@ class ConvertSubprocessor(Subprocessor):
self.frames_idxs = [ *range(len(self.frames)) ] self.frames_idxs = [ *range(len(self.frames)) ]
self.frames_done_idxs = [] self.frames_done_idxs = []
digits = [ str(i) for i in range(10)]
for i in range( len(self.frames) ): for i in range( len(self.frames) ):
frame = self.frames[i] frame = self.frames[i]
frame.idx = i frame.idx = i
frame.output_filename = self.output_path / ('%.5d.png' % (i+1) )
inp_stem = Path(frame.frame_info.filename).stem
if len([ True for symbol in inp_stem if symbol not in digits ]) > 0:
frame.output_filename = self.output_path / ('%.5d.png' % (i+1) )
else:
frame.output_filename = self.output_path / ( inp_stem + '.png' )
frames[0].cfg = self.converter_config.copy() frames[0].cfg = self.converter_config.copy()