当学习深度学习的神经网络时,学到用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

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐