github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-16 10:03:41 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

11
facelib/S3FDExtractor.py
View file

@ -37,16 +37,15 @@ class S3FDExtractor(object):
#filtering faces < 40pix by any side
#enlarging bottom line a bit for 2DFAN-4, because default is not enough covering a chin
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
if min(r-l,bt) < 40:
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),
int(face[1]*input_scale),
int(face[2]*input_scale),
int(face[3]*input_scale)) for face in new_detected_faces ]
return new_detected_faces
def refine(self, olist):
bboxlist = []