light-hub-65 / config.py
ItsMpilo's picture
Deploy Gradio app with multiple files
359c2e8 verified
Raw
History Blame Contribute Delete
906 Bytes
"""
Configuration settings for the video character replacement application
"""
# Model configurations
MEDIAPIPE_MODEL_SELECTION = 0
MEDIAPIPE_MIN_DETECTION_CONFIDENCE = 0.5
# MTCNN configurations
MTCNN_IMAGE_SIZE = 224
MTCNN_MARGIN = 20
MTCNN_MIN_FACE_SIZE = 100
MTCNN_THRESHOLDS = [0.6, 0.7, 0.7]
MTCNN_FACTOR = 0.709
# Processing configurations
DEFAULT_REPLACEMENT_STRENGTH = 0.8
DEFAULT_DETECTION_SENSITIVITY = 0.6
DEFAULT_TRACKING_STABILITY = 0.7
# Video processing
OUTPUT_VIDEO_CODEC = 'mp4v'
PREVIEW_FRAMES_COUNT = 4
# File handling
MAX_FILE_SIZE_MB = 500
SUPPORTED_IMAGE_FORMATS = ['.jpg', '.jpeg', '.png', '.bmp', '.tiff']
SUPPORTED_VIDEO_FORMATS = ['.mp4', '.avi', '.mov', '.mkv', '.wmv']
# Face detection
FACE_DETECTION_OVERLAP_THRESHOLD = 0.5
FACE_MASK_SIGMA = 15
# Color matching
COLOR_MATCH_ENABLED = True
# Performance
MAX_CONCURRENT_PROCESSES = 2
PROCESSING_CHUNK_SIZE = 30 # frames