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

Extractor: image size for all faces now 512

Browse source
This commit is contained in:
Colombo 2020-02-27 16:20:37 +04:00
parent 59be114485
commit 810dc5ada5
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

@ -261,7 +261,7 @@ class ExtractSubprocessor(Subprocessor):
shutil.copy ( str(filepath), str(output_filepath) ) shutil.copy ( str(filepath), str(output_filepath) )
else: else:
output_filepath = output_path / f"{filepath.stem}_{face_idx}.jpg" output_filepath = output_path / f"{filepath.stem}_{face_idx}.jpg"
cv2_imwrite(output_filepath, face_image, [int(cv2.IMWRITE_JPEG_QUALITY), 100] ) cv2_imwrite(output_filepath, face_image, [int(cv2.IMWRITE_JPEG_QUALITY), 90] )
DFLJPG.embed_data(output_filepath, face_type=FaceType.toString(face_type), DFLJPG.embed_data(output_filepath, face_type=FaceType.toString(face_type),
landmarks=face_image_landmarks.tolist(), landmarks=face_image_landmarks.tolist(),
@ -687,7 +687,7 @@ def main(detector=None,
): ):
face_type = FaceType.fromString(face_type) face_type = FaceType.fromString(face_type)
image_size = 512 if face_type == FaceType.WHOLE_FACE else 256 image_size = 512
if not input_path.exists(): if not input_path.exists():
io.log_err ('Input directory not found. Please ensure it exists.') io.log_err ('Input directory not found. Please ensure it exists.')