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

upd SampleProcessor

Browse source
This commit is contained in:
Colombo 2020-07-03 19:31:21 +04:00
parent b0ad36de94
commit 11dc7d41a0
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
samplelib/SampleProcessor.py
View file

@ -6,6 +6,7 @@ import cv2
import numpy as np import numpy as np
from core import imagelib from core import imagelib
from core.cv2ex import *
from core.imagelib import sd from core.imagelib import sd
from facelib import FaceType, LandmarksProcessor from facelib import FaceType, LandmarksProcessor
@ -95,6 +96,7 @@ class SampleProcessor(object):
sample_type = opts.get('sample_type', SPST.NONE) sample_type = opts.get('sample_type', SPST.NONE)
channel_type = opts.get('channel_type', SPCT.NONE) channel_type = opts.get('channel_type', SPCT.NONE)
resolution = opts.get('resolution', 0) resolution = opts.get('resolution', 0)
nearest_resize_to = opts.get('nearest_resize_to', None)
warp = opts.get('warp', False) warp = opts.get('warp', False)
transform = opts.get('transform', False) transform = opts.get('transform', False)
motion_blur = opts.get('motion_blur', None) motion_blur = opts.get('motion_blur', None)
@ -219,6 +221,9 @@ class SampleProcessor(object):
out_sample = np.repeat ( np.expand_dims(cv2.cvtColor(img, cv2.COLOR_BGR2GRAY),-1), (3,), -1) out_sample = np.repeat ( np.expand_dims(cv2.cvtColor(img, cv2.COLOR_BGR2GRAY),-1), (3,), -1)
# Final transformations # Final transformations
if nearest_resize_to is not None:
out_sample = cv2_resize(out_sample, (nearest_resize_to,nearest_resize_to), interpolation=cv2.INTER_NEAREST)
if not debug: if not debug:
if normalize_tanh: if normalize_tanh:
out_sample = np.clip (out_sample * 2.0 - 1.0, -1.0, 1.0) out_sample = np.clip (out_sample * 2.0 - 1.0, -1.0, 1.0)