一种烟盒的包装图像生成方法及电子设备

本申请提供一种烟盒的包装图像生成方法及电子设备，属于计算机图像处理技术领域。方法包括：获取条件数据，条件数据包括用于描述期望生成的包装图像的特征内容的文本数据；将条件数据输入经过训练的图像生成模型，得到由图像生成模型输出的烟盒的至少一个包装图像，其中，图像生成模型包括稳定扩散模型和改进ControlNet模型，在改进ControlNet模型中的可训练副本的每类编码器中设置有改进注意力模块，改进注意力模块包括通道注意力单元、空间注意力单元及深度卷积单元。如此，利用改进注意力模块和ControlNet模型结合，有利于减少训练的参数，能够改善烟盒包装图像的设计效率低的问题。