擅长:python、mysql、java
<p>您可以编写一个自定义生成器类,从目录中读取文件并应用标签。该自定义生成器还可以接受ImageDataGenerator实例,该实例将使用flow()生成批处理。</p>
<p>我在想象这样的事情:</p>
<pre><code>class Generator():
def __init__(self, X, Y, img_data_gen, batch_size):
self.X = X
self.Y = Y # Maybe a file that has the appropriate label mapping?
self.img_data_gen = img_data_gen # The ImageDataGenerator Instance
self.batch_size = batch_size
def apply_labels(self):
# Code to apply labels to each sample based on self.X and self.Y
def get_next_batch(self):
"""Get the next training batch"""
self.img_data_gen.flow(self.X, self.Y, self.batch_size)
</code></pre>
<p>然后简单地说:</p>
<pre><code>img_gen = ImageDataGenerator(...)
gen = Generator(X, Y, img_gen, 128)
model.fit_generator(gen.get_next_batch(), ...)
</code></pre>
<p>*免责声明:我还没有实际测试过,但在理论上应该有效。</p>