开发者社区 > 龙蜥操作系统 > 正文

ModelScope可以再帮我问问卡证检测矫正模型怎么导出onnx格式吗?

ModelScope可以再帮我问问卡证检测矫正模型怎么导出onnx格式吗?

展开
收起
胡青铜 2023-08-14 18:31:18 114 0
1 条回答
写回答
取消 提交回答
  • 可以使用 ExportOnnx 方法导出模型。例如,以下代码可以将卡证检测矫正模型导出为 onnx 格式:

    import modelscope as ms
    from modelscope.models.classification.card_and_idcard_correction import CardAndIdcardCorrectionModel

    创建卡证检测矫正模型

    model = CardAndIdcardCorrectionModel(
    input_shape=(256, 256, 3),
    label_shape=(4,),
    )

    加载模型权重

    model.load_weights("model.h5")

    导出模型为 onnx 格式

    onnx_model = model.export_onnx("model.onnx")
    导出的 onnx 模型可以使用 onnxruntime 等模型运行库执行推理。

    2023-09-25 16:21:47
    赞同 展开评论 打赏

龙蜥社区(OpenAnolis)是面向国际的 Linux 服务器操作系统开源根社区及创新平台,秉承“平等、开放、协作、创新”的原则,理事会由阿里云、统信软件、龙芯、Arm 、Intel 等 24 家国内外头部企业共同组成,有超过 1000 家来自芯片厂商、软件厂商、整机厂商、操作系统厂商等覆盖操作系统全产业链的合作伙伴参与生态共建。

相关电子书

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