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