david/DeepFaceLab_Linux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

S3FD: wrong small face check fix

Browse Source
This commit is contained in:
iperov
2019-03-18 00:30:08 +04:00
parent 2b40fa182f
commit 00e56299c4
1 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
facelib/S3FDExtractor.py
+5 -6
View File
@@ -37,16 +37,15 @@ class S3FDExtractor(object):
#filtering faces < 40pix by any side #filtering faces < 40pix by any side
#enlarging bottom line a bit for 2DFAN-4, because default is not enough covering a chin #enlarging bottom line a bit for 2DFAN-4, because default is not enough covering a chin
new_detected_faces = [] new_detected_faces = []
for l,t,r,b in detected_faces: for ltrb in detected_faces:
l,t,r,b = [ x*input_scale for x in ltrb]
bt = b-t bt = b-t
if min(r-l,bt) < 40: if min(r-l,bt) < 40:
continue continue
new_detected_faces.append ((l,t,r,b+bt*0.1)) b += bt*0.1
new_detected_faces.append ( [int(x) for x in (l,t,r,b) ] )
return [ (int(face[0]*input_scale), return new_detected_faces
int(face[1]*input_scale),
int(face[2]*input_scale),
int(face[3]*input_scale)) for face in new_detected_faces ]
def refine(self, olist): def refine(self, olist):
bboxlist = [] bboxlist = []