一种基于胶囊网络改进的轻量级肺部图像分类方法
摘要:
本发明公开了一种基于胶囊网络改进的轻量级肺部图像分类方法,其借助轻量级肺部图像分类网络模型对肺部图像进行分类处理得到分类识别结果;该轻量级肺部图像分类网络模型包括大核深度可分离卷积模块、感兴趣区域提取模块、分组卷积特征提取模块和胶囊网络模块,借助大核深度可分离卷积模块减少大量的参数量和计算量的同时,提取更多的初步特征信息;借助感兴趣区域提取模块进一步降低模型的计算复杂度和参数量,并准确提取到肺部图像中的感兴趣区域;通过分组卷积特征提取模块进行感兴趣区域的局部区域特征提取,提高模型的表达能力和泛化性能;最后借助胶囊网络模块识别得到分类结果。本发明方法具有参数量和计算量低、分类准确度高的优点。
0/0