mirror of
https://github.com/hacksider/Deep-Live-Cam.git
synced 2026-09-02 05:50:44 +02:00
Tym Rabchuk
9e1f0cc3a5
Review feedback on #1831: - Remove *.gif from the save/output dialog filter (PR had added it there). Verified empirically that cv2.imread/imwrite cannot decode OR encode GIF on OpenCV 4.10 *or* 4.11 (write raises, read returns None), so GIF silently failed on both ends — dropped from every dialog and from has_image_extension. - has_image_extension now uses os.path.splitext so only the true extension counts ('photo.png.bak' / 'clip.webp.mp4' are no longer treated as images). - Centralize the supported-extension set in modules.globals (IMAGE_EXTENSIONS / VIDEO_EXTENSIONS); file_types, all QFileDialog filters and has_image_extension now derive from it instead of hand-copied lists that had already drifted. WEBP itself is unchanged and works (libwebp ships with opencv-python).
85 lines
3.6 KiB
Python
85 lines
3.6 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")
|
|
|
|
# Canonical media extensions, defined once so file dialogs, file_types and
|
|
# has_image_extension never drift. GIF is intentionally excluded: OpenCV's
|
|
# cv2.imread/imwrite (the only image I/O this app uses) cannot decode or
|
|
# encode GIF on 4.10 or 4.11, so offering it would silently fail. WEBP works
|
|
# via the libwebp bundled with opencv-python.
|
|
IMAGE_EXTENSIONS = (".png", ".jpg", ".jpeg", ".bmp", ".webp")
|
|
VIDEO_EXTENSIONS = (".mp4", ".mkv")
|
|
|
|
file_types = [
|
|
("Image", tuple(f"*{ext}" for ext in IMAGE_EXTENSIONS)),
|
|
("Video", tuple(f"*{ext}" for ext in VIDEO_EXTENSIONS)),
|
|
]
|
|
|
|
# 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)
|
|
mouth_mask_size: float = 0.0 # Mouth mask size (0-100; 0=off, 100=mouth to chin)
|
|
|
|
# --- 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 ---
|
|
|
|
import threading
|
|
dml_lock = threading.Lock()
|