AGHRNet 轻量分割网络

精度同优体积更小

专为枣树振动采收设计的深度学习模型,用注意力机制在保持分割精度的同时大幅压缩模型体积,可用于嵌入式视觉系统实时识别树干与树枝。

Zheng, Zhouzhou · Hu, Yaohua · Taifeng, Guo · Qiao, Yichen · He, Yong · 张岩 · Huang, Yuxiang

Computers and Electronics in Agriculture 2023

参数信息

分割精度(mIoU)
77.79 %

技术优势

模型更小

模型大小仅 9.53 MB,比对比方法都小,便于嵌入到移动设备中。

精度更高

mIoU 达到 77.79%,mPA 达到 89.46%,均高于其他十种先进分割方法。

类别更均衡

设计的混合损失函数解决了树干、树枝和背景的像素级类别不平衡问题。

应用场景