如果您正在使用MindSpore 1.2及以上版本,可以改用以下代码导入resnet50模型:

import mindspore.nn as nn
from mindspore import load_checkpoint, Tensor
from mindspore.ops import operations as P

class ResNet50(nn.Cell):
    def __init__(self):
        super(ResNet50, self).__init__()
        self.resnet50 = resnet50(class_num=1000)
        self.softmax = nn.Softmax(axis=1)

    def construct(self, x):
        x = self.resnet50(x)
        x = self.softmax(x)
        return x

# 加载预训练模型参数
checkpoint_path = "resnet50.ckpt"
param_dict = load_checkpoint(checkpoint_path)

# 创建模型并加载参数
net = ResNet50()
net.set_train(False)
net.load_parameters(param_dict)

# 使用模型进行推理
input_data = Tensor(np.zeros([1, 3, 224, 224]), mindspore.float32)
output = net(input_data)
from mindsporemodelzoo import resnet50无法这样导入可以换个方式吗cant find reference model_zoo

原文地址: https://www.cveoy.top/t/topic/bnvM 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录