VLAlert / lkalert /__init__.py
AsianPlayer's picture
Add VLAlert code
1e05592 verified
Raw
History Blame Contribute Delete
526 Bytes
"""
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',
]