From a0e9bdba6e260a9970f228b63ff1f9d66284ad3b Mon Sep 17 00:00:00 2001 From: seranus Date: Fri, 19 Nov 2021 21:33:45 +0100 Subject: [PATCH] sample processor - custom always goes --- samplelib/SampleProcessor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/samplelib/SampleProcessor.py b/samplelib/SampleProcessor.py index 26a5497..74dd18d 100644 --- a/samplelib/SampleProcessor.py +++ b/samplelib/SampleProcessor.py @@ -173,7 +173,7 @@ class SampleProcessor(object): img = imagelib.warp_by_params (params_per_resolution[resolution], img, warp, transform, can_flip=True, border_replicate=border_replicate, cv2_inter=cv2.INTER_LINEAR) img = cv2.resize( img, (resolution,resolution), interpolation=cv2.INTER_LINEAR ) else: - if face_type != sample_face_type: + if face_type != sample_face_type and sample_face_type != FaceType.CUSTOM: # custom always valid for stuff like for wf custom equivivelnet mat = LandmarksProcessor.get_transform_mat (sample_landmarks, resolution, face_type) img = cv2.warpAffine( img, mat, (resolution,resolution), borderMode=borderMode, flags=cv2.INTER_LINEAR ) else: