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

SampleProcessor now uses CUBIC interpolation (was LANCSOZ4) in order to reduce CPU usage

Browse source
This commit is contained in:
iperov 2019-02-03 19:19:42 +04:00
parent 541d6e1cc9
commit 41abda42d2
2 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
samples/SampleProcessor.py
View file

@ -167,9 +167,9 @@ class SampleProcessor(object):
if is_face_sample and target_face_type != -1: if is_face_sample and target_face_type != -1:
if target_face_type > sample.face_type: if target_face_type > sample.face_type:
raise Exception ('sample %s type %s does not match model requirement %s. Consider extract necessary type of faces.' % (sample.filename, sample.face_type, target_face_type) ) raise Exception ('sample %s type %s does not match model requirement %s. Consider extract necessary type of faces.' % (sample.filename, sample.face_type, target_face_type) )
img = cv2.warpAffine( img, LandmarksProcessor.get_transform_mat (sample.landmarks, size, target_face_type), (size,size), flags=cv2.INTER_LANCZOS4 ) img = cv2.warpAffine( img, LandmarksProcessor.get_transform_mat (sample.landmarks, size, target_face_type), (size,size), flags=cv2.INTER_CUBIC )
else: else:
img = cv2.resize( img, (size,size), cv2.INTER_LANCZOS4 ) img = cv2.resize( img, (size,size), cv2.INTER_CUBIC )
if random_sub_size != 0: if random_sub_size != 0:
sub_size = size - random_sub_size sub_size = size - random_sub_size

4
utils/image_utils.py
View file

@ -262,9 +262,9 @@ def gen_warp_params (source, flip, rotation_range=[-10,10], scale_range=[-0.5, 0
def warp_by_params (params, img, warp, transform, flip, is_border_replicate): def warp_by_params (params, img, warp, transform, flip, is_border_replicate):
if warp: if warp:
img = cv2.remap(img, params['mapx'], params['mapy'], cv2.INTER_LANCZOS4 ) img = cv2.remap(img, params['mapx'], params['mapy'], cv2.INTER_CUBIC )
if transform: if transform:
img = cv2.warpAffine( img, params['rmat'], (params['w'], params['w']), borderMode=(cv2.BORDER_REPLICATE if is_border_replicate else cv2.BORDER_CONSTANT), flags=cv2.INTER_LANCZOS4 ) img = cv2.warpAffine( img, params['rmat'], (params['w'], params['w']), borderMode=(cv2.BORDER_REPLICATE if is_border_replicate else cv2.BORDER_CONSTANT), flags=cv2.INTER_CUBIC )
if flip and params['flip']: if flip and params['flip']:
img = img[:,::-1,:] img = img[:,::-1,:]
return img return img