修改Keras ImageDataGenerator和flow_from_directory的类名部分
修改Keras ImageDataGenerator和flow_from_directory的类名部分
提问于 2020-12-07 13:24:08
我想知道是否可以更改flow_from_directory函数，以便给出我想要的class names。
例如，我有名为101, 102的文件夹，而对于flow_from_directory，我将使用y/class names作为101和102，但是对于E 111多标签分类E 212，需要在y中有[1,0,1]和[0,1,0]。
在flow_from_directory中是这样的。
if folder_name == 101:
y=[1,0,1]
if folder_name == 102:
y=[0,1,0]提前谢谢。
回答 1
Stack Overflow用户
回答已采纳
发布于 2020-12-07 15:19:36
如果您试图在flow_from_directory中将一个列表分配给单个类名，您将得到一个错误
对于多标签分类，创建一个包含文件名和标签的熊猫数据，如本例所示，并使用flow_from_dataframe。
然后创建您的列车生成器，将列车dir作为图像的目录。
train_gen = ImageDataGenerator().flow_from_dataframe(
df,
directory=train_dir,
x_col='filename',
y_col='labels',
class_mode='categorical'
)页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/65182547
复制相关文章
相似问题
腾讯云开发者
