基于改进YOLOv8-pose神经网络的速度谱智能拾取方法、介质和设备

本发明涉及一种基于改进YOLOv8‑pose神经网络的速度谱智能拾取方法、介质和设备，属于地球物理勘探技术领域。该方法将传统速度分析问题转化为图像目标与关键点检测，把速度谱中的能量团视为检测目标、把对应的拾取点作为关键点。在保持原有高效检测与关键点回归能力的基础上改进YOLOv8‑pose网络，首先将主干网络部分的SPPF模块替换为SPPELAN模块；其次在检测头部分使用Pose检测头并同时输出能量团的边界框和对应拾取点的坐标。此外，网络训练时采用置信度损失、边界框回归与关键点回归的联合损失。本发明还提了运行所述方法的介质和设备，能够实现高精度、高效率的速度谱拾取。