python - 使用 Keras 进行预处理时出现内存错误

我正在做一个项目，在 Keras 中使用 CNN 对犬种进行分类。在预处理数据时，我收到了这个内存错误：https ://i.stack.imgur.com/e9TkG.png 。我应该怎么办？

标签： pythontensorflowkerasjupyter-notebook

你应该使用data generator.

您收到此错误是因为您的机器没有足够的 RAM 来容纳其上的所有数据。如果您编写生成器，那么您可以获得尽可能多的数据，然后您就可以使用model.fit_generator(). 有了这个，你抓取一些数据，对其进行预处理，并将其转发给模型，抓取另一批数据，对其进行预处理并转发给模型等等。