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

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

展开
收起
青城山下庄文杰 2023-08-11 19:19:52 124 分享 版权
1 条回答
写回答
取消 提交回答
  • 您可以使用 torch.jit.export 函数导出 PyTorch 模型。

    您可以参考以下代码:

    import torch
    import torchvision
    model = torchvision.models.resnet50()
    torch.jit.export(model, "resnet50.onnx")

    2023-09-22 16:26:41
    赞同 展开评论

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

热门讨论

热门文章

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