logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

ControlNet训练自己数据集(数据集处理+环境配置)

image_log 是训练时的测试图例,可以看到生成效果从而判断模型的收敛情况,每次训练都会覆盖此文件夹,建议每训练完一次就改名文件夹防止被覆盖。tutorial_dataset.py 训练和推理都可以共用这个文件,也可以单独再复制一份出来用作读取推理时的condition。lightning_logs 训练完保存到的权重,推理时要把里面的权重路径加上(废话)直接看官方的教程, 按照以下配置去配环

#python#计算机视觉#深度学习 +1
到底了