""" Logging utilities for the YOLO + SAM2 video processing pipeline. """ import logging import sys from typing import Optional def setup_logging(level: str = "INFO", log_file: Optional[str] = None): """ Setup logging configuration. Args: level: Logging level (DEBUG, INFO, WARNING, ERROR) log_file: Optional log file path """ # Convert string level to logging constant numeric_level = getattr(logging, level.upper(), None) if not isinstance(numeric_level, int): raise ValueError(f'Invalid log level: {level}') # Create formatter formatter = logging.Formatter( '%(asctime)s - %(name)s - %(levelname)s - %(message)s' ) # Setup console handler console_handler = logging.StreamHandler(sys.stdout) console_handler.setLevel(numeric_level) console_handler.setFormatter(formatter) # Setup root logger root_logger = logging.getLogger() root_logger.setLevel(numeric_level) root_logger.addHandler(console_handler) # Setup file handler if specified if log_file: file_handler = logging.FileHandler(log_file) file_handler.setLevel(numeric_level) file_handler.setFormatter(formatter) root_logger.addHandler(file_handler) # Reduce noise from some libraries logging.getLogger('ultralytics').setLevel(logging.WARNING) logging.getLogger('matplotlib').setLevel(logging.WARNING) logging.info(f"Logging setup complete - Level: {level}") def get_logger(name: str) -> logging.Logger: """Get a logger instance with the given name.""" return logging.getLogger(name)