一种裂缝提取模型训练方法及装置

本发明涉及油气地质勘探与开发技术领域，公开了一种裂缝提取模型训练方法及装置，可以获取N个露头图像；N为不小于1的整数，每个露头图像中包括裂缝部分和非裂缝部分；对于任一露头图像，根据露头图像中的裂缝部分和非裂缝部分，生成与露头图像对应的灰度图像，将露头图像作为预训练裂缝提取模型的初始训练样本，以及将灰度图像作为初始训练样本对应的初始标签；对多个初始训练样本和对应的初始标签进行数据增强，以构建训练数据集；使用训练数据集对预训练裂缝提取模型进行训练，以得到训练好的裂缝提取模型。本发明训练好的裂缝提取模型可以有效实现在露头图像中进行智能裂缝提取，有效避免相关技术中人工提取存在的需消耗大量人力等缺陷。