github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 04:52:13 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix Util.py, SAE: lowest ae dims now can be 32 (for testing)

Browse source
This commit is contained in:
iperov 2019-02-05 14:43:54 +04:00
parent 0fb3dd53ce
commit 883da4b9ed
2 changed files with 11 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

11
mainscripts/Util.py
View file

@ -29,7 +29,16 @@ def convert_png_to_jpg_file (filepath):
img = cv2.imread (str(filepath))
new_filepath = str(filepath.parent / (filepath.stem + '.jpg'))
cv2.imwrite ( new_filepath, img, [int(cv2.IMWRITE_JPEG_QUALITY), 85])
DFLJPG.embed_data( new_filepath, **dfl_dict )
DFLJPG.embed_data( new_filepath,
face_type=dfl_dict.get('face_type', None),
landmarks=dfl_dict.get('landmarks', None),
yaw_value=dfl_dict.get('yaw_value', None),
pitch_value=dfl_dict.get('pitch_value', None),
source_filename=dfl_dict.get('source_filename', None),
source_rect=dfl_dict.get('source_rect', None),
source_landmarks=dfl_dict.get('source_landmarks', None) )
filepath.unlink()