Yolov5实现物体分类识别和电子围栏

一、程序模块详解

进入电子围栏区域标注:将视频逐帧转化为图片,然后以划线的方式对每张图片进行标注作为进入电子围栏区域的区域。

物体分类识别:基于yolov5模块,对里面的detect.py函数进行修改(更改输入图片的地址并将save_txt置为True,保存识别框的坐标为txt文件。)

坐标转换函数:由于yolov5保存的坐标为标准化后的坐标,而我们所需要的是像素坐标,因此利用此模块将标准化坐标转化为像素坐标。

判断是否跌倒或进入电子围栏区域:在保存的txt文件中可以得到识别物体的分类编号坐标,人的分类编号为0,因此对每张图片把对应的0编号的坐标提取出来,计算矩形框的长宽比c。若c≥1,则代表人跌倒,若c<1则代表人正常行走。标定的进入电子围栏区域可以计算直线的方程,每个人可以得到矩形方框上边沿的终点坐标,带入直线方程即可判断是否进入电子围栏区域,并对其进行标注。

二、功能组成结构图

Yolov5实现物体分类识别和电子围栏_第1张图片

三、实验结果 

Yolov5实现物体分类识别和电子围栏_第2张图片

Yolov5实现物体分类识别和电子围栏_第3张图片 

Yolov5实现物体分类识别和电子围栏_第4张图片 

需要实验报告的小伙伴可以私信我发给你 !!!

你可能感兴趣的:(计算机系大学实验,python,图像处理,开发语言)