修改Keras ImageDataGenerator和flow_from_directory的类名部分
修改Keras ImageDataGenerator和flow_from_directory的类名部分
提问于 2020-12-07 05: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 07: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
复制
相似问题
Keras ImageDataGenerator方法flow_from_directory
Keras ImageDataGenerator和flow_from_directory class_mode='input‘
使用ImageDataGenerator实现Keras中多任务输出的flow_from_directory
如何在keras中使用ImageDataGenerator和flow_from_directory保存增强的图像
如何在Keras中获取由.flow_from_directory函数ImageDataGenerator扫描的类的名称？
添加站长 进交流群
领取专属 10元无门槛券
AI混元助手 在线答疑
关注 腾讯云开发者公众号
洞察 腾讯核心技术
剖析业界实践案例
腾讯云开发者
