<p>你需要一个发电机。参见<a href="https://keras.io/models/sequential/#sequential-model-methods" rel="nofollow noreferrer">^{<cd1>} function section documentation of Keras</a>,或者更好地使用<code>ImageDataGenerator</code>。下一段代码是从<a href="https://github.com/llSourcell/how_to_make_an_image_classifier/blob/master/demo.ipynb" rel="nofollow noreferrer">Building an image classification model using very little data</a>复制粘贴的:</p>
<p><strong>目录结构</strong></p>
<pre><code>data/
train/
dogs/ ### 1024 pictures
dog001.jpg
dog002.jpg
...
cats/ ### 1024 pictures
cat001.jpg
cat002.jpg
...
validation/
dogs/ ### 416 pictures
dog001.jpg
dog002.jpg
...
cats/ ### 416 pictures
cat001.jpg
cat002.jpg
</code></pre>
<p><strong>代码</strong></p>
<pre><code># used to rescale the pixel values from [0, 255] to [0, 1] interval
datagen = ImageDataGenerator(rescale=1./255)
# automagically retrieve images and their classes for train and validation sets
train_generator = datagen.flow_from_directory(
train_data_dir,
target_size=(img_width, img_height),
batch_size=16,
class_mode='binary')
validation_generator = datagen.flow_from_directory(
validation_data_dir,
target_size=(img_width, img_height),
batch_size=32,
class_mode='binary')
</code></pre>
<p>请参阅<a href="https://github.com/llSourcell/how_to_make_an_image_classifier/blob/master/demo.ipynb" rel="nofollow noreferrer">Building an image classification model using very little data</a>中的原始代码。你知道吗</p>