CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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
mainscripts/Extractor.py
+2 -2
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.')