keras使用mnist数据集
安装好tensorflow之后,想通过keras在mnist数据集上跑一下LeNet-5实验,必须得有mnist数据集,这个数据集已经包含在keras中了,在使用的时候会自动下载,具体如下:
>>> from tensorflow.keras.datasets import mnist
2025-10-03 12:50:45.823947: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library cudart64_110.dll
>>> (train_images,train_labels),(test_images,test_labels)=mnist.load_data()
Downloading data from https://storage.googleapis.com/tensorflow/tf-keras-datasets/mnist.npz
11493376/11490434 [==============================] - 3s 0us/step
可以看出,在load_data()时才开始下载,那么下载后的mnist.npz放在哪了?我原以为保存到Anaconda env安装的目录下,结果不是,而是被放在了“C:\Users\[当前登录用户]\.keras\datasets”下面。同样,下载的预训练模型会被放在“C:\Users\[当前登录用户]\.keras\models”下面,比如我下载的resnet50模型“resnet50_weights_tf_dim_ordering_tf_kernels.h5”就保存到这里。
npz是使用numpy读取和保存的格式,h5是HDF5格式。