mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-07-06 21:12:07 -07:00
fix converter
This commit is contained in:
parent
78ddd51ffc
commit
fb4837b3cb
1 changed files with 7 additions and 1 deletions
|
@ -210,10 +210,16 @@ class ConvertSubprocessor(Subprocessor):
|
|||
self.frames_idxs = [ *range(len(self.frames)) ]
|
||||
self.frames_done_idxs = []
|
||||
|
||||
digits = [ str(i) for i in range(10)]
|
||||
for i in range( len(self.frames) ):
|
||||
frame = self.frames[i]
|
||||
frame.idx = i
|
||||
|
||||
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()
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue