我正在使用flow_from_dataframe
将我的图像传递给生成器。但是我使用 cifar10 数据工作。在 dataframe 方法的流程中"y_col"
,需要设置一个参数,其中包含标签列。我的文件如下。
Name 1 2 3 4 5 6 7 8 9 10 11 12 13 14
00001522_000.png 0 0 0 0 0 0 0 0 0 0 0 0 0 0
00023313_000.png 1 0 0 0 0 0 0 0 0 0 0 0 0 0
00023313_001.png 0 0 0 0 0 0 0 0 0 0 0 0 0 0
00023313_002.png 0 0 1 0 0 0 0 1 0 0 …
Run Code Online (Sandbox Code Playgroud)