摘要
本发明涉及药片质量检测技术领域,具体公开了一种基于区域注意机制的药片异常图像生成方法,包括以下步骤:S1、获取药片正常图像数据集和药片异常图像数据集;S2、构建包括生成器和双判别区域注意模块的SDDRAM‑CycleGAN模型;S3、将药片正常图像数据集作为源域,药片异常图像数据集作为目标域,并利用双判别区域注意模块进行双向映射训练,训练过程中通过计算损失函数优化SDDRAM‑CycleGAN模型参数;S4、利用训练好的SDDRAM‑CycleGAN模型,将待转换的药片正常图像生成为异常图像;本发明通过引入双判别区域注意模块和残差引导机制,增强SDDRAM‑CycleGAN模型对异常区域的生成能力,从而实现从正常药片图像到高质量异常图像的精准转换。