天天看点

tf2.中flow和flow_from_directory()区别

tf2.中flow和flow_from_directory()区别

  • Flow()方法要求输入图像X和Y的矩阵,然后根据batch_size输出X和Y,如果要处理的工作数据量较大,则此种方法无法直接全部加载到内存里,因此一般在程序中不常用。
  • Flow_from_directory()方法,它可以迭代的输出X和对应的标签,输入是一个路径,保存格式要求每个类别的图片放在不同的文件夹中,然后会自动建立文件夹名对类名的映射。在深度学习中较为常用

继续阅读