Jeffrey Wu
92c1cda36e
1. encpas the face swapper into a class 2. add node options for switching swapper model
39 lines
1.4 KiB
Python
39 lines
1.4 KiB
Python
import os
|
|
import glob
|
|
import filetype
|
|
|
|
import folder_paths
|
|
|
|
from .typing import FrameFormat, ModelType
|
|
|
|
def is_file(file_path : str) -> bool:
|
|
return bool(file_path and os.path.isfile(file_path))
|
|
|
|
def is_image(image_path : str) -> bool:
|
|
return is_file(image_path) and filetype.helpers.is_image(image_path)
|
|
|
|
|
|
def is_video(video_path : str) -> bool:
|
|
return is_file(video_path) and filetype.helpers.is_video(video_path)
|
|
|
|
def get_temp_frames_pattern(target_path : str, temp_frame_prefix : str, format: FrameFormat) -> str:
|
|
return os.path.join(target_path, temp_frame_prefix + '.' + format)
|
|
|
|
def resolve_relative_path(path : str) -> str:
|
|
return os.path.abspath(os.path.join(os.path.dirname(__file__), path))
|
|
|
|
def get_faceless_models_dir(model_type: ModelType):
|
|
return os.path.join(folder_paths.models_dir,f"faceless/{model_type}")
|
|
|
|
def get_faceless_models(model_type: ModelType):
|
|
models_path = os.path.join(folder_paths.models_dir,f"faceless/{model_type}/*")
|
|
models = glob.glob(models_path)
|
|
models = [x for x in models if x.endswith(".onnx") or x.endswith(".pth")]
|
|
return models
|
|
|
|
def get_faceless_model_path(model_type: ModelType, model_name: str):
|
|
return os.path.join(folder_paths.models_dir,f"faceless/{model_type}/{model_name}")
|
|
|
|
def check_faceless_model_exists(model_type: ModelType, model_name: str):
|
|
return os.path.exists(get_faceless_model_path(model_type, model_name))
|