From e5995630fc02a93d479b258b894f3b6075883e09 Mon Sep 17 00:00:00 2001 From: iperov Date: Sun, 10 Feb 2019 23:00:08 +0400 Subject: [PATCH] fix --- mainscripts/Extractor.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/mainscripts/Extractor.py b/mainscripts/Extractor.py index c58ed1e..5e086aa 100644 --- a/mainscripts/Extractor.py +++ b/mainscripts/Extractor.py @@ -364,16 +364,16 @@ class ExtractSubprocessor(SubprocessorBase): view_rect = (np.array(self.rect) * self.view_scale).astype(np.int).tolist() view_landmarks = (np.array(self.landmarks) * self.view_scale).astype(np.int).tolist() - if self.param_rect_size <= 40: + if self.param_rect_size <= 25: scaled_rect_size = h // 3 if w > h else w // 3 - p1 = (self.param_x - self.param_rect_size // 2, self.param_y - self.param_rect_size // 2) - p2 = (self.param_x + self.param_rect_size // 2, self.param_y - self.param_rect_size // 2) - p3 = (self.param_x - self.param_rect_size // 2, self.param_y + self.param_rect_size // 2) + p1 = (self.param_x - self.param_rect_size, self.param_y - self.param_rect_size) + p2 = (self.param_x + self.param_rect_size, self.param_y - self.param_rect_size) + p3 = (self.param_x - self.param_rect_size, self.param_y + self.param_rect_size) - np1 = (self.param_x - scaled_rect_size // 2, self.param_y - scaled_rect_size // 2) - np2 = (self.param_x + scaled_rect_size // 2, self.param_y - scaled_rect_size // 2) - np3 = (self.param_x - scaled_rect_size // 2, self.param_y + scaled_rect_size // 2) + np1 = (self.param_x - scaled_rect_size, self.param_y - scaled_rect_size) + np2 = (self.param_x + scaled_rect_size, self.param_y - scaled_rect_size) + np3 = (self.param_x - scaled_rect_size, self.param_y + scaled_rect_size) mat = cv2.getAffineTransform( np.float32([p1,p2,p3])*self.view_scale, np.float32([np1,np2,np3])*self.view_scale ) image = cv2.warpAffine(image, mat,(w,h) )