首页> 中文期刊> 《计算机应用与软件》 >基于卷积通道剪枝与多任务的脐橙分级算法

基于卷积通道剪枝与多任务的脐橙分级算法

         

摘要

为实现脐橙品质的快速、准确分级,提出一种同时具有高准确度与高速度的算法。该算法基于高效卷积神经网络(Convolution Neural Network,CNN)设计神经网络结构,结合通道剪枝方法进行推断速度优化。为提升脐橙分级效果,算法采用多任务学习方式,先进行分类,再根据分类结果,让适合回归的种类回归出其瑕疵面积占表面积的比例,为此设计多任务加权损失函数,采用损失函数超参数自适应学习策略,平衡分类与回归的性能。实验表明:该算法在评价指标上取得了最优效果。通道剪枝让模型在CPU下推断速度提升200%,达到90幅/秒的处理速度,准确率仅仅下降2.37百分点。与其他经典模型相比,最好的实验模型准确率可达94.05%。

著录项

相似文献

  • 中文文献
  • 外文文献
  • 专利
获取原文

客服邮箱:kefu@zhangqiaokeyan.com

京公网安备:11010802029741号 ICP备案号:京ICP备15016152号-6 六维联合信息科技 (北京) 有限公司©版权所有
  • 客服微信

  • 服务号