###
计算机系统应用英文版:2023,32(10):45-53
本文二维码信息
码上扫一扫!
改进BiSeNetV1实时模型的岩屑图像识别
(1.四川大学 电子信息学院, 成都 610065;2.成都西图科技有限公司, 成都 610041)
Recognition of Cuttings Images Based on Improved BiSeNetV1 Real-time Model
(1.College of Electronics and Information Engineering, Sichuan University, Chengdu 610065, China;2.Chengdu Xitu Technology Co. Ltd., Chengdu 610041, China)
摘要
图/表
参考文献
相似文献
本文已被:浏览 443次   下载 901
Received:March 13, 2023    Revised:April 20, 2023
中文摘要: 在图像分割识别领域, 现有的深度学习方法大多使用高精度语义分割方法来实现, 存在着网络推理速度慢、计算量大、难以实际应用等问题. 借助于表现较好的BiSeNetV1实时网络模型, 通过扩展的空间路径卷积结构、空间金字塔注意力机制(SPARM)和简化的注意力特征融合模块(S-iAFF)等改进策略, 设计一种用于岩屑图像分割领域的BiSeNet_SPARM_S-iAFF实时网络. 扩展的空间路径卷积结构可以获取更丰富的岩屑图像空间特征, 上下文路径使用优化的空间金字塔注意力机制(SPARM)进一步细化高层语义特征提取, 在特征融合阶段使用简化注意力特征融合(S-iAFF)加强低层空间与高层语义特征的融合程度. 实验结果表明, BiSeNet_SPARM_S-iAFF网络在RockCuttings_Oil岩屑数据集上的平均交并比(mIoU)为64.91%, 相较于BiSeNetV1网络提高了2.68%; 另外改进后的网络在精度上接近大部分高精度语义分割方法, 同时参数量大幅度减少、推理速度有着明显的提升.
Abstract:In the field of image segmentation and identification, the existing deep learning methods mostly perform tasks by high-precision semantic segmentation methods, which lead to a slow network inference speed, large amount of calculation, and difficult actual application. A real-time network model with better performance, namely BiSeNetV1 is used, and the extended spatial path convolution structure, spatial pyramid attention mechanism (SPARM), simplified iterative attention feature fusion (S-iAFF) module, and other optimization strategies are applied. As a result, a real-time BiSeNet_SPARM_S-iAFF network is designed for rock debris image segmentation. The extended spatial path convolution structure can obtain more abundant spatial features of rock debris images. The context path uses the optimized SPARM to further refine high-level semantic feature extraction. Finally, S-iAFF is used to enhance the fusion degree between low-level spatial and high-level semantic features in the feature fusion stage. The experimental results indicate that the mean intersection over union (mIoU) of the BiSeNet_SPARM_S-iAFF network on the RockCuttings_Oil dataset is 64.91%, which is 2.68% higher than that of the BiSeNetV1 network, and the precision of the improved network is close to that of the most high-precision semantic segmentation methods, while the number of parameters is greatly reduced, and the inference speed is significantly improved.
文章编号:     中图分类号:    文献标志码:
基金项目:国家自然科学基金(62071315)
引用文本:
孙杰,滕奇志,罗崇兴,何海波,何小海.改进BiSeNetV1实时模型的岩屑图像识别.计算机系统应用,2023,32(10):45-53
SUN Jie,TENG Qi-Zhi,LUO Chong-Xing,HE Hai-Bo,HE Xiao-Hai.Recognition of Cuttings Images Based on Improved BiSeNetV1 Real-time Model.COMPUTER SYSTEMS APPLICATIONS,2023,32(10):45-53