一种基于语义分割的岩石图像孔隙类型识别方法

摘要

本发明公开了一种基于语义分割的岩石图像孔隙类型识别方法，其包括以下步骤：S1、搭建初始深度学习网络模型；S2、获取岩石原始图像并进行图像切割和图像增强，得到预处理后的图像数据；S3、获取岩石原始图像并进行孔隙位置和形状的人工标注，得到标注后的标签图像数据；S4、对标注后的标签图像数据进行One‑Hot编码，得到编码后的标签数据；S5、将预处理后的图像数据和编码后的标签数据作为训练样本对初始深度学习网络模型进行训练，得到训练后的模型；S6、采用训练后的模型对待识别图像进行识别。本发明抗噪音能力强，具有泛化能力，可提高孔隙识别精度，并且实现了孔隙类别的识别。