1.一种基于改进YOLOv4的图像目标检测方法,其特征在于:基于各幅分别已知其图像中目标类型对象位置的样本图像,按如下步骤A至步骤C,获得目标识别模型;然后针对与样本图像同类型场景的待识别图像,应用目标识别模型获得待识别图像中的目标类型对象位置,完成图像目标检测;
步骤A. 采用预设聚类算法,针对各幅样本图像所构成的图像集进行处理,获得该图像集所对应预设数量M个先验框尺寸,然后进入步骤B;
步骤B. 基于以YOLOv4网络为基础、包括预设数量N个检测层的目标YOLOv4网络,M/N的结果为大于0的整数,按各检测层分别对应M/N个先验框,针对尺寸由小至大排序的各个先验框,顺序分别对应尺度由大至小排序的各检测层,基于目标YOLOv4网络的输入端接收图像,各检测层分别基于其所对应的各先验框、输出该图像所对应的特征图,然后进入步骤C;
步骤C. 基于目标YOLOv4网络、结合其各检测层输出端均对接同一个图像整合模块所构成的待训练识别模型,以及各样本图像,以样本图像作为输入,标识目标类型对象位置的样本图像为输出,针对待训练识别模型进行训练,获得目标识别模型。
2.根据权利要求1所述一种基于改进YOLOv4的图像目标检测方法,其特征在于:所述步骤A中,采用K‑means聚类算法,针对各幅样本图像所构成的图像集进行处理,获得该图像集所对应预设数量M个先验框尺寸。
3.根据权利要求1所述一种基于改进YOLOv4的图像目标检测方法,其特征在于:所述目标YOLOv4网络以YOLOv4网络为基础、包括4个检测层,各检测层分别均包括darknet卷积模块DBL与卷积模块conv,各检测层中,darknet卷积模块DBL的输入端构成检测层的输入端,darknet卷积模块DBL的输出端对接对应卷积模块conv的输入端,卷积模块conv的输出端构成检测层的输出端;各检测层按其尺度由大至小排序,构成检测层排序;
目标YOLOv4网络中还包括前置加强特征提取网络、第一子加强特征提取网络、第二子加强特征提取网络、第三子加强特征提取网络、以及第一前置检测模块、第二前置检测模块、第三前置检测模块;前置加强特征提取网络包括依次串联的darknet卷积模块DBL、上采样并联模块SPP、darknet卷积模块DBL,其中,第一个darknet卷积模块DBL的输入端构成前置加强特征提取网络的输入端,第二个darknet卷积模块DBL的输出端构成前置加强特征提取网络的输出端;各子加强特征提取网络分别均包括上采样模块upsample、连接模块concat、以及三个darknet卷积模块DBL,各子加强特征提取网络中,第一个darknet卷积模块DBL的输入端构成子加强特征提取网络的第一输入端,第二个darknet卷积模块DBL的输入端构成子加强特征提取网络的第二输入端,第一个darknet卷积模块DBL的输出端对接上采样模块upsample的输入端,上采样模块upsample的输出端、第二个darknet卷积模块DBL的输出端对接连接模块concat的输入端,连接模块concat的输出端对接第三个darknet卷积模块DBL的输入端,第三个darknet卷积模块DBL的输出端构成子加强特征提取网络的输出端;第一前置检测模块、第二前置检测模块、第三前置检测模块分别均包括连接模块concat、以及两个darknet卷积模块DBL,各前置检测模块中,第一个darknet卷积模块DBL的输入端构成前置检测模块的第一输入端,第一个darknet卷积模块DBL的输出端对接连接模块concat的其中一个输入端,连接模块concat的另一个输入端构成前置检测模块的第二输入端,连接模块concat的输出端构成前置检测模块的输出端;
目标YOLOv4网络的结构中,YOLOv4网络中主干特征提取网络的输入端构成目标YOLOv4网络的输入端,主干特征提取网络的输出端对接前置加强特征提取网络的输入端,前置加强特征提取网络的输出端对接第一子加强特征提取网络的第一输入端,主干特征提取网络的输出端对接第一子加强特征提取网络的第二输入端;第一子加强特征提取网络的输出端对接第二子加强特征提取网络的第一输入端,主干特征提取网络上顺序第二分支输出端对接第二子加强特征提取网络的第二输入端;第二子加强特征提取网络的输出端对接第三子加强特征提取网络的第一输入端,主干特征提取网络上顺序第一分支输出端对接第三子加强特征提取网络的第二输入端;第三子加强特征提取网络的输出端对接检测层排序中第一个检测层的输入端;第三子加强特征提取网络中连接模块concat的输出端对接第一前置检测模块的第一输入端,第二子加强特征提取网络的输出端对接第一前置检测模块的第二输入端,第一前置检测模块的输出端对接检测层排序中第二个检测层的输入端;第一前置检测模块中连接模块concat的输出端对接第二前置检测模块的第一输入端,第一子加强特征提取网络的输出端对接第二前置检测模块的第二输入端,第二前置检测模块的输出端对接检测层排序中第三个检测层的输入端;第二前置检测模块中连接模块concat的输出端对接第三前置检测模块的第一输入端,前置加强特征提取网络的输出端对接第三前置检测模块的第二输入端,第三前置检测模块的输出端对接检测层排序中第四个检测层的输入端。
4.根据权利要求3所述一种基于改进YOLOv4的图像目标检测方法,其特征在于:所述目标YOLOv4网络中还包括金字塔池化模块,金字塔池化模块包括卷积模块、双线性插值模块、以及预设数量个不同池化核大小的池化模块,其中,各池化模块的输入端构成金字塔池化模块的各输入端,各池化模块的输出端对接卷积模块的输入端,卷积模块的输出端对接双线性插值模块的输入端,双线性插值模块的输出端构成金字塔池化模块的输出端,金字塔池化模块的各输入端同时与金字塔池化模块的输出端相连;所述主干特征提取网络的输出端对接金字塔池化模块的各输入端,金字塔池化模块的输出端对接所述前置加强特征提取网络的输入端。
5.根据权利要求4所述一种基于改进YOLOv4的图像目标检测方法,其特征在于:所述金字塔池化模块中池化模块的数量为4个,分别为6×6池化核大小的池化模块、3×3池化核大小的池化模块、2×2池化核大小的池化模块、1×1池化核大小的池化模块,金字塔池化模块中卷积模块的卷积核大小为1×1。
6.根据权利要求3所述一种基于改进YOLOv4的图像目标检测方法,其特征在于:所述各darknet卷积模块DBL分别均包括顺序相连的卷积模块conv、归一化模块BN、激活模块,其中,卷积模块conv的输入端构成darknet卷积模块DBL的输入端,激活模块的输出端构成darknet卷积模块DBL的输出端。
7.根据权利要求3所述一种基于改进YOLOv4的图像目标检测方法,其特征在于:所述上采样并联模块SPP包括连接模块concat、以及三个最大池化模块maxpool,上采样并联模块SPP中,各最大池化模块maxpool的输入端相连、并构成上采样并联模块SPP的输入端,各最大池化模块maxpool的输出端、以及上采样并联模块SPP的输入端与连接模块concat的输入端相连,连接模块concat的输出端构成上采样并联模块SPP的输出端。
8.根据权利要求3所述一种基于改进YOLOv4的图像目标检测方法,其特征在于:所述检测层排序中第一个检测层对应的尺度为104×104,第二个检测层对应的尺度为52×52,第三个检测层对应的尺度为26×26,第四个检测层对应的尺度为13×13。