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

fixed typo

Browse source
This commit is contained in:
Brigham Lysenko 2019-08-11 15:35:57 -06:00
commit e32c7a963f
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
mainscripts/Extractor.py
View file

@ -243,7 +243,7 @@ class ExtractSubprocessor(Subprocessor):
LandmarksProcessor.draw_rect_landmarks (debug_image, rect, image_landmarks, self.image_size, self.face_type, transparent_mask=True)
if filename_path.suffix == '.jpg':
print(filename_path.suffix + 'is a jpeg')
# if extracting from dflimg and jpg copy it in order not to lose quality
output_file = '{}_{}{}'.format(str(self.final_output_path / filename_path.stem),
str(face_idx), '.jpg')
@ -259,7 +259,7 @@ class ExtractSubprocessor(Subprocessor):
pitch_yaw_roll=data.pitch_yaw_roll
)
else:
print(filename_path.suffix + 'is a png')
output_file = '{}_{}{}'.format(str(self.final_output_path / filename_path.stem),
str(face_idx), '.png')
cv2_imwrite(output_file, face_image, [int(cv2.IMWRITE_PNG_COMPRESSION), 0])