一种基于扩散模型的病理切片虚拟染色的图像生成方法

一种基于扩散模型的病理切片虚拟染色的图像生成方法，属于医疗图像处理技术领域。首先将经过H&E染色作为源图像数据集，将经过IHC染色的病理切片图像作为目标图像数据集；对源图像进行预处理，将源图像转换到对应的YUV颜色空间；提取源图像的显著图和梯度图，生成加权注意力图；利用神经网络搭建扩散模型，输入预处理后的源图像进行微调；反复迭代直到损失函数收敛后，结束训练；输入加权注意力图进行虚拟染色的推理。本发明在推理部分加入病理切片图像的加权注意力作为图像提示输入网络，以控制最终生成图像的细胞结构不被改变，同时保留和强调目标染色图像的颜色特征；提高生成的虚拟染色图像准确率和重建效果。