一种岩石图像分类方法、设备、介质及产品

本发明公开一种岩石图像分类方法、设备、介质及产品，涉及图像分类领域。本发明通过引入特别设计的孪生全局密集网络Siamese Global DenseNet(即SGD模块)，能够减少信息的丢失，并且，SGD模块可以整合全局和局部特征，提高岩石的预测精度。此外，为解决次要特征影响岩石图像的特征分类，本发明还引入了空间转换模块(STN)，以对岩石图像进行变换，实现岩石图像局部关键信息的提取，减少非相关信息对岩石图像分类的影响。通过构建的岩石图像分类模型，基于获取的岩石图像，能够精确地得到岩石图像分类结果，进而提高岩石的预测精度。