以下生成器功能太慢。有没有一种方法可以优化此代码? train_dataset_c1是图像1类别1的火车数据集 train_dataset_c0是图像image,0的Class 0的火车数据集。
<Label Style="{StaticResource highlightedLabel}" Text="I'm Highlighted" />
答案 0 :(得分:2)
如果您使用的是Tensorflow 2.0,我建议您使用tf.data API来加速您的管道。
实际上,您可以将from_generator函数应用于生成器以使其加速
使用此函数将其转换为tf.data.Dataset对象后,您可以使用此tutorial
中的任何策略来对其进行优化