一种基于深度学习的图像虚拟染色方法、装置及相关组件

本发明实施例提供了一种基于深度学习的图像虚拟染色方法、装置及相关组件，该图像虚拟染色方法通过利用经预处理后的光声显微图像和H&E染色图像对预先构建的虚拟染色模型进行迭代训练，得到训练好的虚拟染色模型，再利用训练好的虚拟染色模型对新的光声显微图像进行虚拟染色，得到H&E染色风格的图像(即伪H&E染色图像)。本发明实施例是在生成器中嵌入图像分割模块以帮助生成器区分细胞质与细胞核，从而约束虚拟染色模型，使得生成器更好的生成符合H&E染色图像的形态学特征，以保证生成的伪H&E染色图像中保留了更多有用的细胞核信息，并且预先构建的虚拟染色模型由于在生成器中嵌入图像分割模块，在学习过程中减少了噪声，有效防止过拟合的问题。