热门
ModelScope可以再帮我问问卡证检测矫正模型怎么导出onnx格式吗?
您可以使用 torch.jit.export 函数导出 PyTorch 模型。
您可以参考以下代码:
import torchimport torchvisionmodel = torchvision.models.resnet50()torch.jit.export(model, "resnet50.onnx")
包含图像分类、图像生成、人体人脸识别、动作识别、目标分割、视频生成、卡通画、视觉评价、三维视觉等多个领域