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

fix import

Browse source
This commit is contained in:
Jeremy Hummel 2021-05-23 18:25:25 -07:00
commit ba61e2ffeb
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
samplelib/SampleProcessor.py
View file

@ -7,7 +7,7 @@ import numpy as np
from core import imagelib from core import imagelib
from core.cv2ex import * from core.cv2ex import *
from core.imagelib import sd, blursharpen from core.imagelib import sd, LinearMotionBlur
from core.imagelib.color_transfer import random_lab_rotation from core.imagelib.color_transfer import random_lab_rotation
from facelib import FaceType, LandmarksProcessor from facelib import FaceType, LandmarksProcessor
@ -245,7 +245,7 @@ class SampleProcessor(object):
if blur_type == 'motion': if blur_type == 'motion':
blur_k = np.random.randint(10, 20) blur_k = np.random.randint(10, 20)
blur_angle = 360 * np.random.random() blur_angle = 360 * np.random.random()
img = blursharpen.LinearMotionBlur(img, blur_k, blur_angle) img = LinearMotionBlur(img, blur_k, blur_angle)
elif blur_type == 'gaussian': elif blur_type == 'gaussian':
pass pass