github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-14 02:37:00 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

core.imagelib: update

Browse source
This commit is contained in:
iperov 2021-04-10 09:58:14 +04:00
commit 243f73fafc
5 changed files with 201 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

2
core/imagelib/warp.py
View file

@ -2,7 +2,7 @@ import numpy as np
import cv2
from core import randomex
def gen_warp_params (w, flip, rotation_range=[-10,10], scale_range=[-0.5, 0.5], tx_range=[-0.05, 0.05], ty_range=[-0.05, 0.05], rnd_state=None ):
def gen_warp_params (w, flip=False, rotation_range=[-10,10], scale_range=[-0.5, 0.5], tx_range=[-0.05, 0.05], ty_range=[-0.05, 0.05], rnd_state=None ):
if rnd_state is None:
rnd_state = np.random