一种基于GDS-YOLOv8n的探地雷达图像识别方法

本发明公开了一种探地雷达图像识别方法，特别涉及一种基于GDS‑YOLOv8n的探地雷达图像识别方法，具体步骤包括：步骤1、构建探地雷达图像数据集，数据集划分为训练集、验证集和测试集；步骤2、GDS‑YOLOv8n模型构建，采用DRRB模块、SPD‑Conv模块、Inner‑SIoU和检测头网络构建GDS‑YOLOv8n模型；步骤3、利用GDS‑YOLOv8n模型对构建的探地雷达图像数据集进行识别。将DRRB模块、SPD‑Conv模块、Inner‑SIoU和辅助训练头引入YOLOv8n网络模型中，提出一种GPR图像的识别模型GDS‑YOLOv8n。同时，将经过TensorRT优化后的GDS‑YOLOv8n模型部署到NVIDIA Jetson Orin NX边缘计算设备上，以实现移动端的GPR图像实时解译。本发明不仅在模型推理速度和资源消耗上表现优异，还能够在复杂环境中准确识别目标，满足实际应用中对实时性和便携性的需求。