一种基于迁移学习的钻孔图像岩体结构识别方法

    公开(公告)号:CN119091229A

    公开(公告)日:2024-12-06

    申请号:CN202411286027.1

    申请日:2024-09-13

    Abstract: 本发明提出一种基于迁移学习的钻孔图像岩体结构识别方法,涉及地质勘察技术领域,包括:构建目标检测模型,包括主干网络、区域网络和分类网络;获取第一数据集,进行第一次迭代训练,直至模型收敛,得到第一权重参数;获取第二数据集,根据第一权重参数作为第二次训练的初始参数,进行第二次迭代训练,直至模型收敛,得到第二权重参数;获取第三数据集,根据第二权重参数作为第三次训练的初始参数,进行第三次迭代训练,直至模型收敛,得到最终的目标检测模型;获取钻孔图像,利用最终的目标检测模型对钻孔图像进行识别,得到岩体结构识别结果。本发明基于迁移学习的方法,显著降低了网络对数据集的要求,仅需要较小数据量的数据集即可收敛。

Patent Agency Ranking