计算机技术与发展2024，Vol.34Issue(7)：17-23,7.DOI:10.20165/j.cnki.ISSN1673-629X.2024.0118
基于改进Yolov5s的增强现实文物识别方法
An Augmented Reality Cultural Relics Recognition Method Based on Improved Yolov5s
摘要
Abstract
Applying augmented reality technology to the cultural relics exhibition of the digital museum helps to shorten the distance between the visitors and the cultural relics,and make the exhibition more interesting.Aiming at the problems of false detection and low recognition accuracy caused by complex background and rich shape and texture of cultural relics collected by augmented reality equipment in the application scene of cultural relics exhibition,an improved cultural relics recognition method based on Yolov5s is proposed.The CBAM attention mechanism is introduced between the backbone network and the neck network in the Yolov5s network structure.In addition,a multi-head attention mechanism is used to replace common convolution in the Bottleneck module of the backbone network,which effectively captures local information and reduces the interference of unnecessary information.In order to improve the location accuracy of the boundary frame of the target cultural relics,the DIoU-NMS method is used to select the optimal target identification frame as the final prediction frame.Experimental results show that the proposed algorithm improves the average detection accuracy of the cultural relics,which is more suitable for the target detection of cultural relics than the original algorithm in AR application.关键词
增强现实/文物识别/Yolov5s/注意力机制/多头自注意力机制/DIoU-NMSKey words
augmented reality/cultural relic identification/Yolov5s/attention mechanism/multi-head self-attention/DIoU-NMS分类
信息技术与安全科学引用本文复制引用
张元,关瑜,熊风光,庞敏,况立群..基于改进Yolov5s的增强现实文物识别方法[J].计算机技术与发展,2024,34(7):17-23,7.基金项目
山西省科技重大专项计划"揭榜挂帅"项目(202201150401021) （202201150401021）
