开发者社区 > ModelScope模型即服务 > 计算机视觉 > 正文

基于damoyolo,基于自己的数据集重新训练了一个检测模型,请问如何用pipeline加载?

例如我想加载第24epoch保存的模型并执行预测,有大佬可以指点下如何用pipeline快速加载这个模型并执行预测吗? image.png

展开
收起
dlqdlq 2023-03-10 11:43:15 689 0
1 条回答
写回答
取消 提交回答
  • 达摩院算法工程师

    以DAMO-YOLO-S为例, 1.您可以先调用一遍DAMO-YOLO-S的pipeline, image.png

    1. 找到pipeline推理时的默认缓存路径,如果您是linux系统,该默认路径应该是/home/user_name/.cache/modelscope/hub/damo/cv_tinynas_object-detection_damoyolo
    2. 在该路径下将damoyolo.py文件中的model.name字段,ZeroHead.num_classes字段替换成您的模型。
    3. 再执行一遍步骤1中的推理代码,此时就是调用您自己训练的模型进行推理了。
    2023-03-17 15:09:06
    赞同 展开评论 打赏

包含图像分类、图像生成、人体人脸识别、动作识别、目标分割、视频生成、卡通画、视觉评价、三维视觉等多个领域

相关电子书

更多
低代码开发师(初级)实战教程 立即下载
冬季实战营第三期:MySQL数据库进阶实战 立即下载
阿里巴巴DevOps 最佳实践手册 立即下载