【发布时间】:2018-09-25 14:03:35
【问题描述】:
我正在尝试使用 keras 在 ResNet50 上运行 MNIST 数据集。
我被困在 mnist 数据集上使用 ImageDatagenerator 的部分,因为 keras 可以从目录导入文件。如果我正确理解了data_generator.flow_from_directory,那么它要求图像的路径作为其第一个参数。
但是我无法弄清楚如何输入这些可用的 mnist 数据。
我知道from keras.datasets import mnist 中的可用文件是作为数据框导入的。而flow_from_directory('../input/train',..) 正在询问图像的路径
任何帮助将不胜感激。
from tensorflow.python.keras.applications.resnet50 import preprocess_input
from tensorflow.python.keras.preprocessing.image import ImageDataGenerator
image_size = 224
data_generator = ImageDataGenerator()
import pandas
train_generator = data_generator.flow_from_directory(
'../input/train',
target_size=(image_size, image_size),
batch_size=200,
class_mode='categorical')
validation_generator = data_generator.flow_from_directory(
'../input/val',
target_size=(image_size, image_size),
class_mode='categorical')
【问题讨论】:
标签: python machine-learning keras mnist resnet