lkalert/__init__.py

"""
LKAlert: 基于VLM的主动感知驾驶告警系统
"""

__version__ = "0.1.0"

# 只导入最核心的类，避免循环依赖
from .models.belief_vlm import BeliefActionVLM
from .models.components import TTAHead, PolicyHead

# 配置类
from .utils.config import ModelConfig, TrainingConfig, DataConfig

# 工具函数
from .utils.context import build_context_text

__all__ = [
    'BeliefActionVLM',
    'TTAHead',
    'PolicyHead',
    'ModelConfig',
    'TrainingConfig',
    'DataConfig',
    'build_context_text',
]