当学习深度学习的神经网络时,学到用MNIST图像来训练图像时出现了报错!

# coding: utf-8
import sys, os
sys.path.append(os.pardir)  # 为了导入父目录的文件而进行的设定
import numpy as np
from dataset.mnist import load_mnist
from PIL import Image


def img_show(img):
    pil_img = Image.fromarray(np.uint8(img))
    pil_img.show()

(x_train, t_train), (x_test, t_test) = load_mnist(flatten=True, normalize=False)

img = x_train[0]
label = t_train[0]
print(label)  # 5

print(img.shape)  # (784,)
img = img.reshape(28, 28)  # 把图像的形状变为原来的尺寸
print(img.shape)  # (28, 28)

img_show(img)

运行到时报错了:No module named ‘dataset. mnist’

from dataset.mnist import load_mnist

原因:注意的是此处的dataset并非通过pip install dataset 下载安装的,而是某本书源代码里一个文件夹,mnist是dataset文件夹下的一个py文件,全名mnist.py。你要把dataset文件夹放到你的pycharm或者anaconda编辑器的当前工作目录下,也就lib的工作目录下,我用的是anaconda ,我的工作目录是
在这里插入图片描述
放到lib,下就能成功运行了!
在这里插入图片描述

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐