基于改进YOLOv5s模型的流体包裹体检测算法及其应用

文雪梅 ,  王兴建 ,  宗炜佳 ,  李洋 ,  陈阳 ,  张永恒

科学技术与工程 ›› 2024, Vol. 24 ›› Issue (33) :14122-14128

PDF (10769KB)
科学技术与工程 ›› 2024, Vol. 24 ›› Issue (33) :14122-14128 DOI: 10.12404/j.issn.1671-1815.2400899
天文学、地球科学

基于改进YOLOv5s模型的流体包裹体检测算法及其应用

作者信息 +

Fluid Inclusion Detection Algorithm Based on Improved YOLOv5s Model and Its Application

Author information +
文章历史 +

摘要

流体包裹体对油气资源评价、油藏地球化学、流体类型、流体来源与勘探等都具有重要的指导意义。然而流体包裹体识别主要依赖于人工寻找,这种方法费时费力。为了解决这个问题,提出了一种改进的YOLOv5s流体包裹体目标检测算法。将原始的YOLOv5s模型中的特征提取网络部分和特征融合网络部分进行改进,提高模型的检测能力,使模型更适用于流体包裹体的检测。在特征提取网络部分加入了坐标注意力机制提高定位和识别能力;在特征融合网络部分将原始模型中的路径聚合网络换成双向特征金字塔网络,改进后的网络具有更强大的特征融合能力,可提升小目标的检测能力。通过实验结果表明,与原YOLOv5s模型比较,改进后的YOLOv5s平均精度由75.3%提升到77.3%,比原算法的平均精度提高了2%,检测速度由58.14帧/s帧提升到62.89帧/s,提升了4.75帧/s,实现了更准确高效的流体包裹体检测。

Abstract

Fluid inclusions have important guiding significance for oil and gas resource evaluation, reservoir geochemistry, fluid types, fluid sources, and exploration. However, The identification of fluid inclusions primarily relies on manual searching, a process that is time-consuming and labor-intensive. To address this issue, an improved YOLOv5s fluid inclusion object detection algorithm was proposed. The feature extraction and feature fusion components of the original YOLOv5s model were enhanced to improve the model's detection capability, making it more suitable for fluid inclusion detection. A coordinate attention mechanism was introduced in the feature extraction component to enhance localization and recognition capabilities. Additionally, the original path aggregation network in the feature fusion component was replaced with a bidirectional feature pyramid network. The upgraded network possesses stronger feature fusion capabilities, thereby enhancing the detection capability of small targets. Experimental results demonstrate that compared to the original YOLOv5s model, the average precision of the improved YOLOv5s increases from 75.3% to 77.3%, representing a 2% improvement over the original algorithm. The detection speed also improves from 58.14 frames/s to 62.89 frames/s, resulting in a 4.75 frames/s improvement, thus achieving more accurate and efficient fluid inclusion detection.

关键词

YOLOv5s / 流体包裹体识别 / 目标检测 / 双向特征金字塔网络 / 坐标注意力机制

Key words

YOLOv5s / identification of fluid inclusions / object detection / BiFPN / coordinate attention

引用本文

引用格式 ▾
文雪梅, 王兴建, 宗炜佳, 李洋, 陈阳, 张永恒. 基于改进YOLOv5s模型的流体包裹体检测算法及其应用[J]. 科学技术与工程, 2024, 24(33): 14122-14128 DOI:10.12404/j.issn.1671-1815.2400899
WEN Xue-mei, WANG Xing-jian, ZONG Wei-jia, LI Yang, CHEN Yang, ZHANG Yong-heng. Fluid Inclusion Detection Algorithm Based on Improved YOLOv5s Model and Its Application[J]. Science Technology and Engineering, 2024, 24(33): 14122-14128 DOI:10.12404/j.issn.1671-1815.2400899

登录浏览全文

4963

注册一个新账户 忘记密码

参考文献

基金资助

国家自然科学基金(42074163)

PDF (10769KB)

43

访问

0

被引

详细

导航
相关文章

/

〈 〉