1.一种基于改进PBAS算法与YOLOv5的校园异常行为检测方法,包括以下步骤:
1)使用Yolov5网络进行模型训练:收集大量的数据样本,对样本进行标定,对不同参数设定下的模型进行对比选定最终训练参数和模型;
2)改进PBAS算法提取动态前景:利用改进的PBAS算法对视频运动区域完成动态前景提取,捕捉到有效的动态行为;
3)Yolov5模型检测异常行为:将处理完成后视频帧作为YOLOv5模型的输入进行目标检测,从而确定学生是否存在危险行为;
所述步骤2)具体包括:
2.1)输入视频帧,通过图像的背景模型与当前像素进行,从而确定前景和背景的分类;
其中背景模型是由邻近当前视频帧观察到的N个历史像素值组成的:
B(xi)=B1(xi),...,Bk(xi),...,BN(xi) (1)同时,PBAS算法通过当前帧I(xi)与背景模型B(xi)的比较来决定当前像素属于背景或者前景,具体为比较一个像素点在背景模型中前N个历史像素值与当前像素值,如果当前值与其中至少#min个历史值的距离小于判定阈值R(xi),则将该点判定为前景点,否则将其判定为背景点;
2.2)改进自适应决策阈值更新,根据PBAS算法和步骤2.1)可以确定前景分割掩模计算式为:其中,F=0和F=1分别表示该像素点为背景点和前景点,dist(I(xi),Bk(xi))表示当前点与背景模型的距离;具体改进步骤如下:
2.2.1)首先在算法通过目标框判断运动区域中的像素点为前景点或背景点时,计算出目标框面积为Starget(xi),同时引用一个平衡函数Rban(xi)并设置为固定绝对值;
2.2.2)然后将R(xi)与 由原来的线性关系重新定义为非线性关系;即
其中,定义Rban(xi)为固定绝对值;由于动态背景的目标框面积通常较小,即当Starget(xi),越小像素点xi越有可能为背景点,Rban(xi)与Starget(xi),的比值也就越大,因此与原始PBAS算法中决策阈值R(xi)值相比,改进的决策阈值R(xi)值变化更大,由式(5)可知在进行前景或背景判决时,如果R(xi)值越大, 值小于决策阈值R(xi)的次数大于#min,那么此时像素点就被判别为背景点;而当Starget(xi),超过一定值时,此时运动区域中的像素点更大概率为前景点,Rban(xi)与Starget(xi),的比值也趋向于0,决策阈值R(xi)与背景复杂度仍为线性关系,所以目标框面积Starget(xi),过大并不会影响对运动前景的判断;根据以上分析,可以将决策阈值R(xi)的动态更新方式重新定义为:其中,B(xi)为状态变量,Rban(xi)为固定绝对值;
2.3)更新背景模型,若当前像素点xi被判断为背景点,则用该像素点随机均匀地替换背景样本中的像素点,Bk(xi),k∈1,…,N;在此基础上,当前像素点样本集中的样本点会以1/T(xi)的概率被替换,且每个像素点对应一个概率值1/T(xi);
2.4)判断阈值和更新学习率,引入了背景复杂度使背景决策阈值可以自适应更新,如果一个像素点与背景模型有很大的差异,那么该像素点就会被检测为前景点,并且这个像素点还要以相对较小的概率去更新背景模型;因此PBAS算法在建立背景模型时还建立一个数组D(xi)用来记录相似度最小距离值,而背景复杂度就是相似度最小距离的平均值;
其中, 为N0个最小距离矩阵值的平均值,即背景复杂度;PBAS算法自适应地调整判别阈值如下式所示:其中,Rinde和Rscale均为预先设定的固定值;学习率T(xi)的更新策略表示为:其中,Tinc和Tdec为预先设定的固定值。
2.如权利要求1所述的一种基于改进PBAS算法与YOLOv5的校园异常行为检测方法,所述步骤1)具体包括:
1.1)数据集制作:将收集到的三种异常类型视频进行视频帧抽取;然后将数据集进行图片预处理,对图片进行筛选,滤波去噪,图像增强;最后,使用LabelImg标图工具对预处理后图片中的异常行为区域进行框定,生成txt文件,以供模型训练使用;
1.2)模型训练:经过反复训练模型的尝试,同时结合训练结果和YOLOv5网络自身结构,设定相关参数如下:初始学习率为0.01,学习率衰减权重为0.0005,训练迭代次数为200次;
YOLOv5的损失函数使用GIOU Loss作为bounding box,Box推测出的值即为GIoU损失函数的均值。