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

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

8
mainscripts/Converter.py
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()