基于空间-轴向双重注意力机制的断层识别方法及系统

本发明公开一种基于空间‑轴向双重注意力机制的断层识别方法及系统，该方法包括：采用高精度正演褶积算法构建原始训练数据集；设计融合UNet与ResNet架构的URNet基础网络，并进行初步训练，获得预训练模型；对原始训练数据集进行数据增强与扩充，形成增强训练数据集；基于预训练模型，集成空间‑轴向双重注意力机制，并引入多级残差连接结构，构建包含双向增强型解码器的URNet‑TSA神经网络；利用增强训练数据集对URNet‑TSA神经网络进行二次训练；利用训练好的URNet‑TSA神经网络进行断层识别。本发明通过构建URNet‑TSA新型神经网络架构，提升复杂地质断层特征的捕捉能力以及识别精度。