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