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

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

展开
收起
dlqdlq 2023-03-10 11:43:15 752 分享 版权
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
    赞同 展开评论

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

还有其他疑问?
咨询AI助理