[发明专利]基于YOLOV4模型的违章建筑检测方法

权利要求书

1.一种违章建筑检测方法，其特征在于，包括：

根据预选区域的视频获取违章建筑的图片；

对图片进行标注，并根据标注后的图片获取先验框；

将由图片构成的数据集划分为训练集和测试集；

对训练集进行预处理；

根据预处理后的训练集对YOLOV4模型进行训练；

根据测试集对训练后的YOLOV4模型进行调整；以及

将视频输入调整后的YOLOV4模型以检测视频内的违章建筑。

2.如权利要求1所述的违章建筑检测方法，其特征在于，

所述根据预选区域的视频获取违章建筑的图片的方法包括：

拍摄预选区域的视频，选取视频中带有违章建筑的视频片段，以获得带有违章建筑的图片，并且将图片的分辨率调整至预设分辨率。

3.如权利要求2所述的违章建筑检测方法，其特征在于，

所述对图片进行标注，并根据标注后的图片获取先验框的方法包括：

对分辨率调整后的图片中违章建筑的位置进行标注，获取目标框，获取标记数据中目标框的长和宽以及位置；

以标注的物体类别以及目标框的长和宽，以及位置作为该图片的标签，并对目标框的长和宽进行归一化：

其中，w_r是归一化之后目标框的宽；h_r是归一化之后目标框的高；w为目标框的宽；h为目标框高；W为图片的宽；H为图片的高；

初始化先验框的类别数量和聚类中心，并计算每个目标框与所有聚类中心的距离IOU：

in＝min(h₁,h₂)min(w₁,w₂)；

un＝h₁w₁+h₂w₂；

其中，in为两个目标框的交集；un为两个目标框的并集；h₁为一个目标框的高；w₁为一个目标框的宽；h₂为另一个目标框的高；w₂为另一个目标框的宽；

选取最近的聚类中心为该目标框的类别；

根据每个簇的均值作为下次迭代的聚类中心，直到每个类别的中心位置在相邻两次迭代的误差在ε内，则最后的聚类中心为先验框。

4.如权利要求3所述的违章建筑检测方法，其特征在于，

所述将由图片构成的数据集划分为训练集和测试集的方法包括：

将由分辨率调整后的图片构成的数据集按预设比例划分为训练集和测试集。

5.如权利要求4所述的违章建筑检测方法，其特征在于，

所述对训练集进行预处理的方法包括：

对训练集进行数据增强，即

每次选取训练集中预设张数的图片，对图片进行翻转、缩放、色域变化处理，并将各图片按预设位置摆放，以进行图片的组合和框的组合；

将图片的类别编号编码为one-hot编码，并对标签进行处理。