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

1

Browse source
This commit is contained in:
Colombo 2020-01-22 22:14:31 +04:00
commit f1a3150ce9
6 changed files with 75 additions and 50 deletions
Download patch file Download diff file Expand all files Collapse all files

5
merger/FrameInfo.py
View file

@ -1,9 +1,8 @@
from pathlib import Path
class FrameInfo(object):
def __init__(self, filename=None, landmarks_list=None):
self.filename = filename
self.filename_short = str(Path(filename).name)
def __init__(self, filepath=None, landmarks_list=None):
self.filepath = filepath
self.landmarks_list = landmarks_list or []
self.motion_deg = 0
self.motion_power = 0

2
merger/MergeMasked.py
View file

@ -334,7 +334,7 @@ def MergeMaskedFace (predictor_func, predictor_input_shape, cfg, frame_info, img
def MergeMasked (predictor_func, predictor_input_shape, cfg, frame_info):
img_bgr_uint8 = cv2_imread(frame_info.filename)
img_bgr_uint8 = cv2_imread(frame_info.filepath)
img_bgr_uint8 = imagelib.normalize_channels (img_bgr_uint8, 3)
img_bgr = img_bgr_uint8.astype(np.float32) / 255.0