from .YOLOP import get_net as get_net_yolop
from .YOLOP_YOLOv11 import get_net_yolov11

def get_net(cfg, **kwargs):
    """
    根据配置选择合适的模型
    
    Args:
        cfg: 配置对象
        **kwargs: 额外参数（用于 YOLOv11）
    """
    if hasattr(cfg.MODEL, 'USE_YOLOV11') and cfg.MODEL.USE_YOLOV11:
        return get_net_yolov11(cfg, **kwargs)
    else:
        return get_net_yolop(cfg, **kwargs)