一种基于机器学习的中医耳部五脏区域分割方法和装置

摘要

本发明公开了一种基于机器学习的中医耳部五脏区域分割方法和装置，包括：采集耳部图像，建立图像数据集；对耳部图像进行标注，得到标签图像；分别扩充耳部图像和标签图像；对扩充标签图像进行Canny算子边缘检测；分别对扩充耳部图像、扩充标签图像和标签边缘图像进行预处理；利用灰度图像和二值图像生成心、肝、脾、肺、肾各单个区域边缘图像；将RGB三通道图像输入学习网络，得到预测图像；从预测图像中获得心、肝、脾、肺、肾的各单个区域预测图像；分别对预测图像进行膨胀、腐蚀处理，得到其单区域粗略边缘预测图；计算总损失函数；计算总损失函数的最小值，使学习网络收敛，利用学习网络对耳部图像进行分割，得到耳部五脏区域分割图像。