From b3b97b9802868a12d15e52c78ff472949d0efc24 Mon Sep 17 00:00:00 2001 From: Colombo Date: Tue, 25 Feb 2020 09:59:48 +0400 Subject: [PATCH] fix RuntimeWarning during the extraction process --- mainscripts/Extractor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mainscripts/Extractor.py b/mainscripts/Extractor.py index 69c5a30..7602cc8 100644 --- a/mainscripts/Extractor.py +++ b/mainscripts/Extractor.py @@ -241,8 +241,8 @@ class ExtractSubprocessor(Subprocessor): landmarks_bbox = LandmarksProcessor.transform_points ( [ (0,0), (0,image_size-1), (image_size-1, image_size-1), (image_size-1,0) ], image_to_face_mat, True) - rect_area = mathlib.polygon_area(np.array(rect[[0,2,2,0]]), np.array(rect[[1,1,3,3]])) - landmarks_area = mathlib.polygon_area(landmarks_bbox[:,0], landmarks_bbox[:,1] ) + rect_area = mathlib.polygon_area(np.array(rect[[0,2,2,0]]).astype(np.float32), np.array(rect[[1,1,3,3]]).astype(np.float32)) + landmarks_area = mathlib.polygon_area(landmarks_bbox[:,0].astype(np.float32), landmarks_bbox[:,1].astype(np.float32) ) if not data.manual and face_type <= FaceType.FULL_NO_ALIGN and landmarks_area > 4*rect_area: #get rid of faces which umeyama-landmark-area > 4*detector-rect-area continue