我想亲自训练controlnet,但我发现准备数据集很不方便。正如我在此链接中提供的huggingface教程:https://huggingface.co/blog/train-your-controlnet,我相信我应该以huggingface数据集格式组织数据集。我的目的是使用各种提示设置来训练controlnet并比较结果。然而,我意识到我需要为每个实验创建多个数据集,这是耗时且空间效率低下的,因为每个数据集中的图像和条件图像保持不变。
我应该创建多个仅在提示列中不同的数据集,还是有更有效的方法来实现这一点?
1条答案
按热度按时间ep6jt1vc1#
创建多个只有提示列的数据集(例如
controlnet_prompts_1
、controlnet_prompts_2
等)和一个具有图像、条件图像和除提示列(例如,controlnet_features
)。然后,每当你想使用提示数据集与主特征数据集的特定组合时,使用
concatenate_datasets
连接两个数据集,如下所示(文档):