一种岩石分类方法、终端设备及存储介质

    公开(公告)号:CN114170457A

    公开(公告)日:2022-03-11

    申请号:CN202111329893.0

    申请日:2021-11-11

    Abstract: 本发明涉及一种岩石分类方法、终端设备及存储介质,该方法中包括:S1:采集不同类型的纹理图像组成第一训练集,采集不同类型的岩石图像组成第二训练集;S2:构建基于残差块堆叠网络的分类模型,采用迁移学习的方法,设定第一训练集为迁移学习的源域、第二训练集为迁移学习的目标域,在只改变分类模型中的全连接层的参数的情况下对模型进行训练,将训练后的分类模型作为岩石分类模型;S3:通过岩石分类模型对待识别岩石图像进行类型识别,得到待识别岩石图像对应岩石的类型。本发明通过迁移学习来防止模型过拟合或者欠拟合,提高岩石分类识别效果。

Patent Agency Ranking