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

2019-02-05 14:43:54 +04:00
parent 0fb3dd53ce
commit 883da4b9ed
2 changed files with 11 additions and 2 deletions
@@ -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()