def _get_aviris(self): data = tf.data.Dataset.from_generator(self._aviris_generator, output_types = (tf.float32, tf.float32)) data = data.batch(self.batch_size) data = data.cache() data = data.prefetch(2) data = data.repeat() return data
时间: 2024-03-30 11:36:38 浏览: 70
indianPine.zip_drawm9a_indian pine_indianpines_indianpine数据集_印第安
这段代码是一个函数,函数名为 `_get_aviris`,该函数返回一个 TensorFlow 的数据集对象 `data`。这个数据集对象是通过调用 `tf.data.Dataset.from_generator` 方法创建的,它的数据来源是 `self._aviris_generator` 方法,该方法应该是一个生成器函数。
这个数据集对象 `data` 通过 `data.batch` 方法被分成批次,每个批次大小为 `self.batch_size`。接着,数据集被缓存到内存中,以便下次使用。然后,数据集对象被预取了 2 个批次的数据,以加速训练。最后,数据集对象被重复使用,以支持无限迭代获取数据。
阅读全文