发明 一种岩石图像检索方法及其系统 【特价】
图像检索 地质检测 计算机视觉 1人
G06V10/764 G06V10/774 G06V10/26 G06N3/04 G06N3/08
摘要:本发明公开了一种岩石图像检索方法,包括:实时获取图像数据,将图像数据输入到训练好的深度学习网络模型中,得到检索的结果图;所述述深度学习网络模型的训练过程包括:利用已收集的图像数据构建岩石图像数据集;将数据集输入到网络中,通过空间转移模块处理后,使网络能够主动地在空间上转换特征映射;将处理后的数据输入多粒度网络中,计算模型的总损失函数和mAP值,经过多次计算,当损失函数趋于稳定,且mAP值达到峰值时,完成深度学习网络模型的训练;本发明避免了仅利用表征对岩石图像进行分类,同时对岩石图像的细粒度特征有更精确的提取,有助于在有杂物遮挡、样本数量较少和质量较低、信息缺失等情况下提高岩石图像的检索准确率。