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
在 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])
])
原文地址: https://www.cveoy.top/t/topic/bEod 著作权归作者所有。请勿转载和采集!