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

DFL-2.0 initial branch commit

Browse source
This commit is contained in:
Colombo 2020-01-21 18:43:39 +04:00
parent 52a67a61b3
commit 38b85108b3
154 changed files with 5251 additions and 9414 deletions
Download patch file Download diff file Expand all files Collapse all files

8
samplelib/SampleProcessor.py
View file

@ -4,7 +4,7 @@ from enum import IntEnum
import cv2
import numpy as np
import imagelib
from core import imagelib
from facelib import FaceType, LandmarksProcessor
@ -154,9 +154,9 @@ class SampleProcessor(object):
yaw = -yaw
if img_type == SPTF.IMG_PITCH_YAW_ROLL_SIGMOID:
pitch = (pitch+1.0) / 2.0
yaw = (yaw+1.0) / 2.0
roll = (roll+1.0) / 2.0
pitch = np.clip( (pitch / math.pi) / 2.0 + 1.0, 0, 1)
yaw = np.clip( (yaw / math.pi) / 2.0 + 1.0, 0, 1)
roll = np.clip( (roll / math.pi) / 2.0 + 1.0, 0, 1)
img = (pitch, yaw, roll)
else: