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

ModelScope的pipeline加载自己训练的模型时出了问题怎么办?

ModelScope的pipeline加载自己训练的模型时出了问题怎么办?

展开
收起
青城山下庄文杰 2023-08-12 09:45:19 157 0
1 条回答
写回答
取消 提交回答
  • 您试试下面方法:
    pipeline里的model='xxx' # xxx为保存训练时设置的work_dir在pipeline下加入下面代码

    checkpoint = load_checkpoint(
                        p.model,
                        '/x/xxx/xxxx/xxxxx/best_accuracy_top-1_epoch_1.pth', # work_dir下的best开头的.pth文件
                        map_location='cpu')
    if 'CLASSES' in checkpoint.get('meta', {}):
        CLASSES = checkpoint['meta']['CLASSES']
    p.model.CLASSES=CLASSES
    
    # 后面使用pipeline推理
    p('img_path')
    

    -此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”

    2023-08-15 08:22:59
    赞同 展开评论 打赏

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

相关电子书

更多
视觉AI能力的开放现状及ModelScope实战 立即下载
ModelScope助力语音AI模型创新与应用 立即下载
低代码开发师(初级)实战教程 立即下载