github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-10 15:23:32 -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

13
mainscripts/Extractor.py
View file

@ -79,13 +79,24 @@ class ExtractSubprocessor(Subprocessor):
image = self.cached_image[1]
else:
image = cv2_imread( filename_path_str )
h, w, ch = image.shape
wm = w % 2
hm = h % 2
if wm + hm != 0: #fix odd image
image = image[0:h-hm,0:w-wm,:]
self.cached_image = ( filename_path_str, image )
if image is None:
self.log_err ( 'Failed to extract %s, reason: cv2_imread() fail.' % ( str(filename_path) ) )
else:
if self.type == 'rects':
rects = self.e.extract_from_bgr (image)
h, w, ch = image.shape
if min(w,h) < 128:
self.log_err ( 'Image is too small %s : [%d, %d]' % ( str(filename_path), w, h ) )
rects = []
else:
rects = self.e.extract_from_bgr (image)
return [str(filename_path), rects]
elif self.type == 'landmarks':