github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-07 05:22:06 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

S3FD : filtering faces < 40pix.

Browse source
This commit is contained in:
iperov 2019-03-14 01:23:39 +04:00
parent 8da47fec13
commit 62af1b6f16
3 changed files with 27 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

2
facelib/MTCExtractor.py
View file

@ -90,7 +90,7 @@ class MTCExtractor(object):
input_image = input_image[:,:,::-1].copy()
(h, w, ch) = input_image.shape
input_scale = self.scale_to / (w if w > h else h)
input_scale = self.scale_to / max(w,h)
input_image = cv2.resize (input_image, ( int(w*input_scale), int(h*input_scale) ), interpolation=cv2.INTER_LINEAR)
detected_faces, pnts = detect_face ( input_image, self.min_face_size, self.pnet_fun, self.rnet_fun, self.onet_fun, [ self.thresh1, self.thresh2, self.thresh3 ], self.scale_factor )