mirror of
https://github.com/deepfakes/faceswap.git
synced 2026-10-18 14:34:30 +08:00
torzdf
43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
#!/usr/bin/env python3
|
|
""" Parent class for scaling Adjustments for faceswap.py converter """
|
|
|
|
import logging
|
|
import numpy as np
|
|
|
|
from lib.logger import parse_class_init
|
|
from plugins.convert import convert_config
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class Adjustment():
|
|
""" Parent class for scaling adjustments """
|
|
def __init__(self, config_file=None):
|
|
logger.debug(parse_class_init(locals()))
|
|
convert_config.load_config(config_file=config_file)
|
|
logger.debug("Initialized %s", self.__class__.__name__)
|
|
|
|
def process(self, new_face):
|
|
""" Override for specific scaling adjustment process """
|
|
raise NotImplementedError
|
|
|
|
def run(self, new_face):
|
|
""" Perform selected adjustment on face """
|
|
# pylint:disable=duplicate-code
|
|
logger.trace("Performing scaling adjustment") # type:ignore[attr-defined]
|
|
# Remove Mask for processing
|
|
reinsert_mask = False
|
|
final_mask = None
|
|
if new_face.shape[2] == 4:
|
|
reinsert_mask = True
|
|
final_mask = new_face[:, :, -1]
|
|
new_face = new_face[:, :, :3]
|
|
new_face = self.process(new_face)
|
|
new_face = np.clip(new_face, 0.0, 1.0)
|
|
if reinsert_mask and new_face.shape[2] != 4:
|
|
# Reinsert Mask
|
|
assert final_mask is not None
|
|
new_face = np.concatenate((new_face, np.expand_dims(final_mask, axis=-1)), -1)
|
|
logger.trace("Performed scaling adjustment") # type:ignore[attr-defined]
|
|
return new_face
|