github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-16 10:03:41 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix converter issue #205

Browse source
This commit is contained in:
iperov 2019-03-26 18:23:39 +04:00
parent 070e938568
commit 009af42617
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
mainscripts/Converter.py
View file

@ -54,7 +54,12 @@ class ConvertSubprocessor(Subprocessor):
if self.converter.type == Converter.TYPE_FACE and filename_path.stem not in self.alignments.keys():
if not self.debug:
self.log_info ( 'no faces found for %s, copying without faces' % (filename_path.name) )
shutil.copy ( str(filename_path), str(output_filename_path) )
if filename_path.suffix == '.png':
shutil.copy ( str(filename_path), str(output_filename_path) )
else:
image = cv2_imread(str(filename_path))
cv2_imwrite ( str(output_filename_path), image )
else:
image = (cv2_imread(str(filename_path)) / 255.0).astype(np.float32)