为了掌握使用ControlNet与StableDiffusion,训练了一个包含7600张飞鸟图像的自定义数据集,使用canny控制进行了5个周期的训练,使用A100 GPU。结果很有前景。GitHub链接如下,有更多详细信息。https://github.com/anilbhatt1/Deeplearning-E8P1/tree/master/Controlnet-canny