[发明专利]基于遥感影像快速构建飞机分类数据集的方法

权利要求书

1.基于遥感影像快速构建飞机分类数据集的方法，其特征在于，包括：

步骤1：标注原始遥感影像，用矩形框标注所述原始遥感影像中的飞机目标，并记录标注框在影像中的坐标，形成原始标注文件；

步骤2：根据原始标注文件初次裁剪原始遥感影像，形成每张切片只包含一个飞机目标的切片，切片尺寸及位置通过飞机目标所在标注框的坐标计算得到，同时生成与每张切片相对应的切片标注文件；

步骤3：旋转切片，并计算旋转后飞机目标在旋转后切片中的坐标，然后根据该坐标从旋转后的切片中裁剪飞机目标；

步骤2中，初次裁剪原始遥感影像，按单个目标从原始遥感影像裁剪，形成每张切片只包含一个飞机目标的切片，且飞机目标位于切片中心，切片尺寸及位置通过飞机目标所在标注框的坐标计算得到，同时生成与每张切片相对应的标注文件；

步骤2包括：

(1)裁剪一个飞机目标：

记原始遥感影像的宽度W，高度H，读取飞机目标所在的标注框的位置为(x1,y1),(x2,y2),(x3,y3),(x4,y4)，计算左上角坐标(x_min,y_min)和右下角坐标(x_max,y_max)：

x_min＝min(x1,x2,x3,x4)

y_min＝min(y1,y2,y3,y4)

x_max＝max(x1,x2,x3,x4)

y_max＝max(y1,y2,y3,y4)

计算飞机目标的宽width和高height：

width＝x_max-x_min

height＝y_max-y_min

计算待裁剪切片的大小，计算切片需要扩充的大小extend：

extend＝max(width,height)/2

计算待裁剪切片在原始遥感影像中的坐标，左上角(x_c_min,y_c_min)和右下角(x_c_max,y_c_max)，并按此坐标从原始遥感影像中裁剪切片，(x_c_min,y_c_min)和(x_c_max,y_c_max)坐标计算公式如下：

x_c_min＝max(0,x_min-extend)

y_c_min＝max(0,y_min-extend)

x_c_max＝min(W,x_max+extend)

y_c_max＝min(H,y_max+extend)

计算该飞机目标在切片中的坐标，左上角为(x_n_min,y_n_min)，右下角为(x_n_max,y_n_max)，并生成与该切片对应的切片标注文件，所述切片标注文件中包含飞机目标的类型和飞机目标相对于切片的坐标，(x_n_min,y_n_min)和(x_n_max,y_n_max)坐标计算公式如下：

x_n_min＝x_min–x_c_min

y_n_min＝y_min–y_c_min

x_n_max＝x_n_min+width

y_n_max＝y_n_min+height

(2)裁剪所有飞机目标

按照(1)中的步骤将原始遥感影像中的所有飞机目标逐个裁剪，生成独立切片和对应的切片标注文件。

2.根据权利要求1所述的基于遥感影像快速构建飞机分类数据集的方法，其特征在于，步骤1中原始标注文件记录飞机目标的类型和飞机目标所在的标注框在所述原始遥感影像中的坐标。

3.根据权利要求2所述的基于遥感影像快速构建飞机分类数据集的方法，其特征在于，步骤3中对步骤2中生成的切片进行多个角度旋转，并从旋转后的切片中裁剪飞机目标，生成数据集，包括：

步骤3.1：旋转切片获得新图片，计算以新图片的左上角为原点时原始切片中任一像素点对应的旋转坐标；

步骤3.2：计算飞机目标在新图片中标注框的四个点的坐标；

步骤3.3：按正方形裁剪飞机目标，获得正方形飞机目标图像；

步骤3.4：循环执行步骤3.1～3.3，获得其他旋转角度的正方形飞机目标图像和其他切片多个旋转角度的正方形飞机目标图像，生成数据集。