基于改进Unet的矿石图像分割
DOI:
作者:
作者单位:

1.河南理工大学物理与电子信息学院 焦作 454000;2.无锡东如科技有限公司工业智能研究院 无锡 214000

作者简介:

通讯作者:

中图分类号:

TP391

基金项目:


Ore image segmentation based on improved Unet
Author:
Affiliation:

1.School of Physics and Electronic Information, Henan Polytechnic University,Jiaozuo 454000,China; 2.Wuxi Dongru Technology Co., Ltd.,Institute of Industrial Intelligence,Wuxi 214000,China

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    为了应对矿产领域的矿石识别困难、识别设备成本高等问题提出了改进的Unet矿石图像分割算法,首先对EfficientNetV2加以改造作为模型的主干网络提取矿石特征;其次,引入MBconv模块作为解码器中的基本模块,加强特征提取能力;采用CA注意力模块替换SE注意力模块,保留更多空间位置信息;最后使用滤波器响应归一化(FRN)层替换常用的批量归一化(BN)层,避免模型性能受批量大小的影响。基于FeM和Cu数据集实验结果表明,PA分别为96.58%和95.39%,Miou分别为92.8%和90.43%,F1_score分别为95.15%和93.47%,Efficient_unet模型参数相比Unet减少了60.4%,推理速度提升了19.23%,可达21.7 fps,所提模型在精度和速率方面均优于现有经典分割模型,同时表现出良好的泛化性能。

    Abstract:

    Addressing the challenges of ore recognition in the mining industry and the high cost of recognition equipment, we propose an improved Unet ore image segmentation algorithm. Firstly, we modify EfficientNetV2 as the backbone network of the model to extract ore features. Secondly, we introduce the MBconv module as the decoder, enhancing the feature extraction capability. We then replace the SE attention module with the CA attention module to retain more spatial position information. Finally, we substitute the commonly used Batch Normalization (BN) layer with the Filter Response Normalization (FRN) layer to prevent model performance from being affected by batch size. Experimental results based on FeM and Cu datasets demonstrate that our proposed model achieves a PA of 96.58% and 95.39%, an MIoU of 92.8% and 90.43%, and an F1 score of 95.15% and 93.47%. Compared to Unet, the Efficient_Unet model parameters are reduced by 60.4%, and the inference speed is improved by 19.23%, reaching 21.7 frames per second. Our proposed model outperforms existing classical segmentation models in terms of accuracy and speed, exhibiting strong generalization performance.

    参考文献
    相似文献
    引证文献
引用本文

曾中华,曹东.基于改进Unet的矿石图像分割[J].电子测量技术,2023,46(21):176-182

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-03-04
  • 出版日期: