This commit is contained in:
Colombo 2019-11-16 16:54:10 +04:00
parent b74c2b1f9a
commit 668513a507
2 changed files with 2 additions and 2 deletions

View file

@ -242,7 +242,7 @@ class ExtractSubprocessor(Subprocessor):
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] )
if landmarks_area > 4*rect_area: #get rid of faces which umeyama-landmark-area > 4*detector-rect-area
if self.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
if self.debug_dir is not None: