一种热轧过程的钢板图像分割方法及装置

本申请公开了一种热轧过程的钢板图像分割方法及装置，涉及图像分割技术领域，利用SAM大模型对热轧过程图像中的钢板区域进行标记，得到分割更为精准的二值化标签图像，并将原始钢板图像与二值化标签图像作为分割模型的输入数据和相应标签，从而构建用于热轧过程钢板图像的数据集，能够对热轧过程中钢板图像进行准确的标注，解决图像数据标注效率和准确率低的问题。随后，通过使用图像增强方式提高对热轧过程中钢板图像数据的多样性，提升训练模型的鲁棒性和准确性。然后，通过引入注意力机制模块构建钢板图像分割模型，实现对热轧过程中钢板图像准确地分割和轮廓特征提取，解决钢板温度变化导致的钢板颜色变化和水雾遮挡等问题。