mirror of
https://github.com/hacksider/Deep-Live-Cam.git
synced 2026-04-29 04:45:59 +02:00
Lauri Gates
e57116de68
Add two new face enhancement processors using GPEN-BFR ONNX models at 256x256 and 512x512 resolutions. Models auto-download on first use from GitHub releases. Integrates into existing frame processor pipeline alongside GFPGAN enhancer with UI toggle switches. - modules/paths.py: Shared path constants module - modules/processors/frame/_onnx_enhancer.py: ONNX enhancement utilities - modules/processors/frame/face_enhancer_gpen256.py: GPEN-BFR 256 processor - modules/processors/frame/face_enhancer_gpen512.py: GPEN-BFR 512 processor - modules/core.py: Add GPEN choices to --frame-processor CLI arg - modules/globals.py: Add GPEN entries to fp_ui toggle dict - modules/ui.py: Add GPEN toggle switches and processing integration Closes #1663 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
73 lines
3.0 KiB
Python
73 lines
3.0 KiB
Python
# --- START OF FILE globals.py ---
|
|
|
|
import os
|
|
from typing import List, Dict, Any
|
|
|
|
ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
WORKFLOW_DIR = os.path.join(ROOT_DIR, "workflow")
|
|
|
|
file_types = [
|
|
("Image", ("*.png", "*.jpg", "*.jpeg", "*.gif", "*.bmp")),
|
|
("Video", ("*.mp4", "*.mkv")),
|
|
]
|
|
|
|
# Face Mapping Data
|
|
source_target_map: List[Dict[str, Any]] = [] # Stores detailed map for image/video processing
|
|
simple_map: Dict[str, Any] = {} # Stores simplified map (embeddings/faces) for live/simple mode
|
|
|
|
# Paths
|
|
source_path: str | None = None
|
|
target_path: str | None = None
|
|
output_path: str | None = None
|
|
|
|
# Processing Options
|
|
frame_processors: List[str] = []
|
|
keep_fps: bool = True
|
|
keep_audio: bool = True
|
|
keep_frames: bool = False
|
|
many_faces: bool = False # Process all detected faces with default source
|
|
map_faces: bool = False # Use source_target_map or simple_map for specific swaps
|
|
poisson_blend: bool = False # Enable Poisson Blending for smoother face swaps
|
|
color_correction: bool = False # Enable color correction (implementation specific)
|
|
nsfw_filter: bool = False
|
|
|
|
# Video Output Options
|
|
video_encoder: str | None = None
|
|
video_quality: int | None = None # Typically a CRF value or bitrate
|
|
|
|
# Live Mode Options
|
|
live_mirror: bool = False
|
|
live_resizable: bool = True
|
|
camera_input_combobox: Any | None = None # Placeholder for UI element if needed
|
|
webcam_preview_running: bool = False
|
|
show_fps: bool = False
|
|
|
|
# System Configuration
|
|
max_memory: int | None = None # Memory limit in GB? (Needs clarification)
|
|
execution_providers: List[str] = [] # e.g., ['CUDAExecutionProvider', 'CPUExecutionProvider']
|
|
execution_threads: int | None = None # Number of threads for CPU execution
|
|
headless: bool | None = None # Run without UI?
|
|
log_level: str = "error" # Logging level (e.g., 'debug', 'info', 'warning', 'error')
|
|
|
|
# Face Processor UI Toggles (Example)
|
|
fp_ui: Dict[str, bool] = {"face_enhancer": False, "face_enhancer_gpen256": False, "face_enhancer_gpen512": False}
|
|
|
|
# Face Swapper Specific Options
|
|
face_swapper_enabled: bool = True # General toggle for the swapper processor
|
|
opacity: float = 1.0 # Blend factor for the swapped face (0.0-1.0)
|
|
sharpness: float = 0.0 # Sharpness enhancement for swapped face (0.0-1.0+)
|
|
|
|
# Mouth Mask Options
|
|
mouth_mask: bool = False # Enable mouth area masking/pasting
|
|
show_mouth_mask_box: bool = False # Visualize the mouth mask area (for debugging)
|
|
mask_feather_ratio: int = 12 # Denominator for feathering calculation (higher = smaller feather)
|
|
mask_down_size: float = 0.1 # Expansion factor for lower lip mask (relative)
|
|
mask_size: float = 1.0 # Expansion factor for upper lip mask (relative)
|
|
|
|
# --- START: Added for Frame Interpolation ---
|
|
enable_interpolation: bool = True # Toggle temporal smoothing
|
|
interpolation_weight: float = 0 # Blend weight for current frame (0.0-1.0). Lower=smoother.
|
|
# --- END: Added for Frame Interpolation ---
|
|
|
|
# --- END OF FILE globals.py ---
|