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

ModelScope这个bin文件可以用pytorch读取使用吗?

ModelScope这个bin文件可以用pytorch读取使用吗?
86b9898776d5fb3b137af5a101977953.png

展开
收起
青城山下庄文杰 2023-08-05 20:59:38 149 0
1 条回答
写回答
取消 提交回答
  • ModelScope 的 bin 文件是 TorchScript 格式的模型文件,可以用 torch.jit.load 方法加载。

    以下是一个使用 torch.jit.load 加载 TorchScript 格式模型文件的示例代码:

    import torch

    model = torch.jit.load("model.bin")

    input = torch.rand(1, 3, 224, 224)

    output = model(input)
    更多关于 TorchScript 格式模型文件的使用,请参考 官方文档。

    2023-09-25 17:34:04
    赞同 展开评论 打赏

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

相关电子书

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