/
githubmirror
/
faceswap
Обзор
Документация
Войти
/
githubmirror
/
faceswap
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
plugins/convert/scaling/_base.py
42 строки
1 KB
torzdf
Extraction Overhaul (#1533)
21 мар 2026, 02:52
Не верифицирован
21 мар 2026, 02:52
d217812
Код
Авторство
О чём код?
#!/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