一种轻量型平衡胶囊网络模型的瓷器碎片分类方法

    公开(公告)号:CN117541866A

    公开(公告)日:2024-02-09

    申请号:CN202311519087.9

    申请日:2023-11-14

    申请人: 西北大学

    摘要: 本发明公开了一种轻量型平衡胶囊网络模型的瓷器碎片分类方法,包括如下步骤:步骤1、采集和构建瓷器碎片图像数据集;步骤2、对瓷器碎片图像进行预处理;步骤3、先在胶囊网络的普通卷积层和主胶囊层之间引入一个轻量化通道变换模块ChannelTrans,再结合焦点损失focal loss设计损失函数MF‑R loss,然后利用动态学习率对原始动态路由进行自适应改进,从而构建出用于对陶瓷碎片进行分类的轻量型平衡胶囊网络模型;步骤4、整合并训练轻量型平衡胶囊网络模型;步骤5、利用训练好的轻量型平衡胶囊网络模型预测出图像分类结果,提高了对陶瓷图像分类的准确率。