[发明专利]自适应铁路货车异常检测训练集图像增强方法

权利要求书

1.自适应铁路货车异常检测训练集图像增强方法，其特征在于，所述方法包括：

S1、读取铁路货车异常检测训练集中的一张图像；

S2、对图像进行灰度处理；

S3、计算灰度处理后图像中的像素平均亮度值，根据平均亮度值判断是否符合亮度需求，若否，将灰度处理后图像作为原始图像，对原始图像进行亮度增强，将亮度增强后的图像补充到训练集中，转入S4，若是，转入S4；

S4、读取训练集中的下一张图像，转入S2，直至处理完训练集中需要亮度增强的图像；

S3中，进行亮度增强的方法包括：

根据标注区域，选取原始图像中的目标区域，对目标区域进行高斯亮度变化：

其中，A为变换后的目标区域，I为原始图像，为对于目标区域中像素点(x,y)的高斯变换结果，T为图像变换方式；

x_c,y_c,w,h分别为目标区域检测框的中心点横坐标、中心点纵坐标、检测框的长度、检测框的高度；b为预设亮度值，mean为像素平均亮度值，H、W分别为原始图像的高、宽，r表示检测框的面积与原始图像面积的比值。

2.根据权利要求1所述的自适应铁路货车异常检测训练集图像增强方法，其特征在于，所述S4中，直至处理完训练集中需要亮度增强的图像，转入S5；

S5、对训练集中的图像进行缩放，包括：

根据标注区域，选取亮度增强后的图像中的目标区域，计算目标区域占原始图像整体的面积比例z，Area_bbox表示目标区域的面积，Area_image表示原始图像的面积；

若z≥t，则将目标区域缩小至原有的p倍，0＜p≤0.5，图像尺寸不变，缩小后周围用0填充，将图像中标注区域的坐标信息做相应的变化，保证缩小前后的标注区域不变；

若z＜t，则将目标区域放大至原有的p倍，1＜p≤k，其中k为原始图像尺寸的长与目标区域尺寸的长的比值与原始图像尺寸的宽与目标区域尺寸的宽的比值中的最大值，将图像中标注区域的坐标信息做相应的变化，保证放大前后的标注区域不变；

t表示预设阈值。

3.一种计算机可读的存储设备，所述存储设备存储有计算机程序，其特征在于，所述计算机程序被执行时实现如权利要求1至2任一所述自适应铁路货车异常检测训练集图像增强方法。

4.一种自适应铁路货车异常检测训练集图像增强装置，包括存储设备、处理器以及存储在所述存储设备中并可在所述处理器上运行的计算机程序，其特征在于，所述处理器执行所述计算机程序实现如权利要求1至2任一所述自适应铁路货车异常检测训练集图像增强方法。