一种基于深度卷积对抗生成网络的皮肤病图像的生成方法技术

一种基于深度卷积对抗生成网络的皮肤病图像的生成方法，包括以下步骤：步骤1：对原始训练数据的预处理，对原始训练数据进行分割，得到皮肤病变区域的分割图片；步骤2：深度卷积对抗生成网络结构的建立，使用卷积神经网络来进行分别构建生成器和判别器；步骤3：设定生成网络和分类网络的输入；步骤4：优化目标的设定，对于判别器来说，判别器负责对真假样本进行分类，生成器生成的样本为假，实际输入的真实样本为真，最小化分类误差；对于生成器来说，生成器尽可能的去生成和真实样本一样的假样本；步骤5：网络的训练。本发明专利技术利用人工分割出来的皮损样本作为对抗生成网络的训练样本，然后使用对抗生成网络来生成新的皮肤病样本。

【技术实现步骤摘要】
一种基于深度卷积对抗生成网络的皮肤病图像的生成方法
本专利技术涉及计算机视觉领域。本专利技术公开了一种使用对抗生成网络来生成皮肤病新样本的方法。具体先人工将皮肤病的损伤区域分割出来，然后利用分割出来的图像作为对抗生成网络的训练样本来进行新样本的生成。
技术介绍
近年来，由于深度学习在处理图像的分类、检测等问题上优于传统的机器学习方法，因此得到了极大的发展，但是因为深度学习优越的性能是建立在大量训练数据的基础上的，所以收集到更多的训练数据显得尤为重要。由于皮肤病图像可收集的病患样本有限，并且很多数据处于不公开状态，所以为了解决皮肤图像样本量少的问题，本专利技术提出了一种使用对抗生成网络来生成皮肤病图像的方法。对抗生成网络是Goodfellow在2014年提出的一种生成式模型。系统由一个生成器和一个判别器构成，生成器捕捉真实数据样本的潜在分布，判别器是一个二分类器，判别输入的是真实数据还是生成的样本。训练时固定一方，更新另一个模型的参数，交替迭代，使得对方的错误最大化，最终能估测出样本数据的分布。目前对抗生成网络最常使用的地方就是图像生成，如超分辨率任务，语义分割等。
技术实现思路
为了解决皮肤病图像样本量不足的问题，本专利技术提供一种基于深度卷积对抗生成网络的皮肤病图像的生成方法，该方法利用人工分割出来的皮损样本作为对抗生成网络的训练样本，然后使用对抗生成网络来生成新的皮肤病样本。为了解决上述技术问题，本专利技术提供如下的技术方案：一种基于深度卷积对抗生成网络的皮肤病图像的生成方法，所述生成方法包括以下步骤：步骤1：对原始训练数据的预处理，对原始训练数据进行分割，得到皮肤病变区域的分割图片；步骤2：深度卷积对抗生成网络结构的建立，使用卷积神经网络来进行分别构建生成器和判别器；步骤3：设定生成网络和分类网络的输入，将随机产生的符合均匀分布的随机噪声数据作为步骤2中构建的生成器的输入，将训练后的生成器的输出和步骤1中得到的皮肤病变区域的分割图片样本作为步骤2中构建的判别器的输入；步骤4：优化目标的设定，对于判别器来说，判别器负责对真假样本进行分类，生成器生成的样本为假，实际输入的真实样本为真，最小化分类误差；对于生成器来说，生成器尽可能的去生成和真实样本一样的假样本；步骤5：网络的训练：先对生成器网络进行训练，训练周期为n个epoch,然后固定生成器，并将其输出和步骤1中分割出来的皮肤病变区域图片作为判别器网络的输入然后训练2n个epoch。循环按照1:2的训练次数训练生成器和判别器。进一步，所述步骤1中，通过分割出皮肤的病变区域，分割具体步骤如下：1.1、依次从原始训练集中取训练数据样本；1.2、设定训练样本的坐标规则，左上角为起始坐标(0,0)右下角坐标为结束坐标；1.3、使用标注工具对于原始皮肤图片样本中的病变区域进行标注得到其左上角起始坐标和右下角结束坐标，如果同一张图片存在多个非粘连在一起的皮肤病变，则分别对各个区域进行分割；1.4、根据1.3中得到的坐标将该区域分割出来并对分割出来的皮肤病变区域图片按照规则重新命名。再进一步，所述的步骤3中，生成器和判别器的建立过程为：生成器的建立：生成器一共由5层结构组成，其中前4层对应的结构组成为二维反卷积层+BatchNorm层+Relu激活层。第5层其结构由二维反卷积层+Tanh激活层，其中所有的二维反卷积层的卷积核大小为4×4，不同之处在于，第1层的卷积步长为1，填充为0，第2、3、4、5层的卷积步长为2，填充为1；判别器的建立：判别器一共由5层结构组成，其中第1层的组成结构为二维卷积层+LeakyReLu激活层，第2、3、4层的组成结构为二维卷积层+BatchNorm层+LeakyReLu激活层，第5层的组成结构为二维卷积层+Sigmoid激活层；激活层函数表达式：ReLu激活函数：x代表输入，f(x)代表输出。LeakyReLu激活函数：其中x代表输入，f(x)代表输出，α是一个人为设定的固定值；Sigmoid激活函数：其中x代表输入，f(x)代表输出。本专利技术的有益效果为：利用人工分割出来的皮损样本作为对抗生成网络的训练样本，然后使用对抗生成网络来生成新的皮肤病样本。附图说明图1为本专利技术中分割皮肤病图像示意图；图2为对抗生成网络整体原理框架图图3为生成器网络的结构示意图；图4为判别器网络的结构示意图；图5为整个生成模型最终的生成皮肤病图像和真实图像的对比图，其中，(a)为生成图，(b)为真实图像。具体实施方式以下将参照附图，对本专利技术的优选实施例进行详细的描述。应当理解，优选实施例仅为了说明本专利技术，而不是为了限制本专利技术的保护范围。参照图1～图5，一种基于深度卷积对抗生成网络的皮肤病图像的生成方法，所述生成方法包括以下步骤：步骤1：对原始训练数据的预处理，对原始训练数据进行分割，得到皮肤病变区域的分割图片；步骤2：深度卷积对抗生成网络结构的建立，使用卷积神经网络来进行分别构建生成器和判别器；步骤3：设定生成网络和分类网络的输入，将随机产生的符合均匀分布的随机噪声数据作为步骤2中构建的生成器的输入，将训练后的生成器的输出和步骤1中得到的皮肤病变区域的分割图片样本作为步骤2中构建的判别器的输入；步骤4：优化目标的设定，对于判别器来说，判别器负责对真假样本进行分类，生成器生成的样本为假，实际输入的真实样本为真，最小化分类误差；对于生成器来说，生成器尽可能的去生成和真实样本一样的假样本；步骤5：网络的训练：先对生成器网络进行训练，训练周期为n个epoch,然后固定生成器，并将其输出和步骤1中分割出来的皮肤病变区域图片作为判别器网络的输入然后训练2n个epoch。循环按照1:2的训练次数训练生成器和判别器。进一步，所述步骤1中，通过分割出皮肤的病变区域，分割具体步骤如下：1.1、依次从原始训练集中取训练数据样本；1.2、设定训练样本的坐标规则，左上角为起始坐标(0,0)右下角坐标为结束坐标；1.3、使用标注工具对于原始皮肤图片样本中的病变区域进行标注得到其左上角起始坐标和右下角结束坐标，如果同一张图片存在多个非粘连在一起的皮肤病变，则分别对各个区域进行分割；1.4、根据1.3中得到的坐标将该区域分割出来并对分割出来的皮肤病变区域图片按照规则重新命名。再进一步，所述的步骤3中，生成器和判别器的建立过程为：生成器的建立：生成器一共由5层结构组成，其中前4层对应的结构组成为二维反卷积层+BatchNorm层+Relu激活层。第5层其结构由二维反卷积层+Tanh激活层，其中所有的二维反卷积层的卷积核大小为4×4，不同之处在于，第1层的卷积步长为1，填充为0，第2、3、4、5层的卷积步长为2，填充为1；判别器的建立：判别器一共由5层结构组成，其中第1层的组成结构为二维卷积层+LeakyReLu激活层，第2、3、4层的组成结构为二维卷积层+BatchNorm层+LeakyReLu激活层，第5层的组成结构为二维卷积层+Sigmoid激活层；激活层函数表达式：ReLu激活函数：x代表输入，f(x)代表输出。LeakyReLu激活函数：其中x代表输入，f(x)代表输出，α是一个人为设定的固定值；Sigmoid激活函数：其中x代表输入，f(x)代表输出。本实施例的基于深度卷积对抗生成网络的皮本文档来自技高网...

【技术保护点】
1.一种基于深度卷积对抗生成网络的皮肤病图像的生成方法，其特征在于，所述生成方法包括以下步骤：步骤1：对原始训练数据的预处理，对原始训练数据进行分割，得到皮肤病变区域的分割图片；步骤2：深度卷积对抗生成网络结构的建立，使用卷积神经网络来进行分别构建生成器和判别器；步骤3：设定生成网络和分类网络的输入，将随机产生的符合均匀分布的随机噪声数据作为步骤2中构建的生成器的输入，将训练后的生成器的输出和步骤1中得到的皮肤病变区域的分割图片样本作为步骤2中构建的判别器的输入；步骤4：优化目标的设定，对于判别器来说，判别器负责对真假样本进行分类，生成器生成的样本为假，实际输入的真实样本为真，最小化分类误差；对于生成器来说，生成器尽可能的去生成和真实样本一样的假样本；步骤5：网络的训练：先对生成器网络进行训练，训练周期为n个epoch,然后固定生成器，并将其输出和步骤1中分割出来的皮肤病变区域图片作为判别器网络的输入然后训练2n个epoch。循环按照1:2的训练次数训练生成器和判别器。

【技术特征摘要】
1.一种基于深度卷积对抗生成网络的皮肤病图像的生成方法，其特征在于，所述生成方法包括以下步骤：步骤1：对原始训练数据的预处理，对原始训练数据进行分割，得到皮肤病变区域的分割图片；步骤2：深度卷积对抗生成网络结构的建立，使用卷积神经网络来进行分别构建生成器和判别器；步骤3：设定生成网络和分类网络的输入，将随机产生的符合均匀分布的随机噪声数据作为步骤2中构建的生成器的输入，将训练后的生成器的输出和步骤1中得到的皮肤病变区域的分割图片样本作为步骤2中构建的判别器的输入；步骤4：优化目标的设定，对于判别器来说，判别器负责对真假样本进行分类，生成器生成的样本为假，实际输入的真实样本为真，最小化分类误差；对于生成器来说，生成器尽可能的去生成和真实样本一样的假样本；步骤5：网络的训练：先对生成器网络进行训练，训练周期为n个epoch,然后固定生成器，并将其输出和步骤1中分割出来的皮肤病变区域图片作为判别器网络的输入然后训练2n个epoch。循环按照1:2的训练次数训练生成器和判别器。2.如权利要求1所述的一种基于深度卷积对抗生成网络的皮肤病图像的生成方法，其特征在于：所述步骤1中，分割出皮肤的病变区域，分割具体流程如下：1.1、依次从原始训练集中取训练数据样本；1.2、设定训练样本的坐标规则，左上角为起始坐标(0,0)右下角坐标为结束坐标；...

【专利技术属性】
技术研发人员：胡海根，孔祥勇，苏一平，陈胜勇，管秋，周乾伟，
申请(专利权)人：浙江工业大学，
类型：发明
国别省市：浙江,33