yetrun's picture
ver2: 扩展 CV 训练框架,支持分类、分割与目标检测任务
14f6839
Raw
History Blame Contribute Delete
250 Bytes
"""
图像分割模型入口。
这里暴露 Oxford Pets 分割任务使用的模型构建函数,供流水线模型阶段调用。
"""
from models.segmentation.model_builder import build_segmentation_model
__all__ = ["build_segmentation_model"]