Yu Chen
> 请问可以生成彩色的缺陷图片吗？想从彩色原图中把划痕，污染，异物这种不良剪切出来进行生成类似的不良。 可以生成彩色的
你可以考虑一下stable-diffusion，那个开源生成模型支持多种sample😃
类别就是可以生成不同的图片，训练时按照类别训练，生成时按照类别生成图片。
目前没有做这方面工作
现在阶段的版本不推荐使用修改image_size生成大尺寸图像，建议开启`--latent`，加载autoencoder和diffusion模型权重进行大尺寸生成。 具体训练办法请参考：[https://github.com/chairc/Integrated-Design-Diffusion-Model/blob/main/docs/zh-Hans/02_%E8%AE%AD%E7%BB%83.md](https://github.com/chairc/Integrated-Design-Diffusion-Model/blob/main/docs/zh-Hans/02_%E8%AE%AD%E7%BB%83.md)
是是因为selfattention那个模块吃显存，你可以参考一下unet_slim.py代码。关于大尺寸训练，建议加载latent训练权重训练64*64。如果直接训练256不启用latent，建议调小batch size与num worker，启用amp，切换unet_slim模型。