在 MindSpore 中,图像变换的组合不再使用 py_transforms.Compose,而是使用 mindspore.dataset.transforms 中的 Compose。因此,需要将以下代码段:

from mindspore.dataset.vision import py_transforms

transform = py_transforms.Compose([
    py_transforms.ToTensor(),
    py_transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
])

修改为:

from mindspore.dataset.transforms import Compose, Normalize, ToTensor

transform = Compose([
    ToTensor(),
    Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
])
import cv2import numpy as npimport mindsporefrom mindspore import Tensor load_checkpoint load_param_into_netfrom mindsporedatasetvision import py_transformsfrom PIL import Imagefrom main import ResNet

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

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