收录:
摘要:
本发明公开了一种基于轻型卷积神经网络和知识蒸馏的中医舌图像自动分割方法,设计了一种轻型密集连接卷积模块,并利用该模块构建轻型舌图像分割网络;利用舌图像分割数据集训练网络,并采用知识蒸馏的方法增强网络的准确性和鲁棒性,得到分割网络模型。具体包括以下步骤:舌图像分割数据集的构建;轻型舌图像分割网络的设计;基于知识蒸馏的轻型舌图像分割网络的训练。与现有技术相比本发明模型参数小、分割速度快。采用了知识蒸馏的训练方法,提出的分割网络在分割精度和鲁棒性方面都获得与教师模型可以比较的结果,满足实际应用的需求。
关键词:
通讯作者信息:
电子邮件地址: