代码:img = Image.open("test.png").convert('L') img2 = img.resize((28, 28), Image.ANTIALIAS) img_array = np.array(img2),在软件pytorch报错Expected type 'Union[ndarray, Iterable, int, float]', got 'Image' instead,如何解决报错?或给出替换代码
时间: 2023-06-04 19:06:59 浏览: 136
和矩阵相关的头文件matrix.h
5星 · 资源好评率100%
您好!对于您的问题,您可以将代码修改为以下形式:
img = Image.open("test.png").convert('L')
img2 = img.resize((28, 28), Image.ANTIALIAS)
img_array = np.array(img2)
# 将 Image 对象转换为 ndarray 对象
img_array = np.asarray(img_array)
这段代码将 Image 对象转换为 ndarray 对象,以避免您在使用 PyTorch 时遇到的错误。希望能对您有所帮助!
阅读全文