-
公开(公告)号:CN114170457A
公开(公告)日:2022-03-11
申请号:CN202111329893.0
申请日:2021-11-11
Applicant: 厦门理工学院
IPC: G06V10/764 , G06V10/82 , G06V10/774 , G06K9/62 , G06N3/04 , G06N3/08
Abstract: 本发明涉及一种岩石分类方法、终端设备及存储介质,该方法中包括:S1:采集不同类型的纹理图像组成第一训练集,采集不同类型的岩石图像组成第二训练集;S2:构建基于残差块堆叠网络的分类模型,采用迁移学习的方法,设定第一训练集为迁移学习的源域、第二训练集为迁移学习的目标域,在只改变分类模型中的全连接层的参数的情况下对模型进行训练,将训练后的分类模型作为岩石分类模型;S3:通过岩石分类模型对待识别岩石图像进行类型识别,得到待识别岩石图像对应岩石的类型。本发明通过迁移学习来防止模型过拟合或者欠拟合,提高岩石分类识别效果。