一种基于SAM分割模型的红外图像样本增广方法

本发明提供基于SAM分割模型的红外图像样本增广方法，属于红外图像目标检测领域，生成待增广目标标注坐标列表；调用SAM分割模型和权重依次分割目标，生成多个目标掩模；从原图中分别获得对应位置的像素值，生成多个目标的分割结果图；对每个目标区域分别计算二值化阈值并二值化，得到二值图；腐蚀操作，如果模拟当前光照范围扩大，则对光线照射区进行膨胀操作；模拟当前光照范围缩小，将光线照射区重新赋值为0，将待照射区重新赋值为255，膨胀操作；生成大于1的系数一和小于1的系数二，将目标掩模范围内变亮区域的像素值依次乘以系数一，将目标掩模范围内变暗区域的像素值依次乘以系数二，将计算结果大于255的像素值赋值为255；同时模拟并扩增同一图像的多个目标在不同光照角度下的形态，扩充样本集，增加模型泛化性。