(19)国家知识产权局
(12)发明 专利申请
(10)申请公布号
(43)申请公布日
(21)申请 号 20221087646 6.2
(22)申请日 2022.07.25
(71)申请人 中原动力 智能机器人有限公司
地址 450018 河南省郑州市河南自贸试验
区郑州片区(郑东)三全路与龙源四街
交叉口中原科技城创新孵化基地C座2
层
(72)发明人 袁野 万里红 吴国栋 刘娜
(74)专利代理 机构 广州三环 专利商标代理有限
公司 44202
专利代理师 吴松滨
(51)Int.Cl.
G06T 7/73(2017.01)
G06V 40/10(2022.01)
G06V 40/20(2022.01)G06V 10/25(2022.01)
G06V 10/62(2022.01)
G06V 10/82(2022.01)
(54)发明名称
一种社交距离风险预警方法及装置
(57)摘要
本发明公开了一种社交距离风险预警方法
及装置, 所述方法包括: 将待检测图片输入至行
人检测模型中, 以使行人检测模 型根据待检测图
片得到多个行人的边界框信息; 其中, 待检测图
片中存在多个行人; 根据边界框信息, 结合BYTE
跟踪算法, 计算得到多个行人id信息; 根据行人
id信息计算多个行人坐标, 对多个行人坐标依次
进行逆透视变换, 分别生成多个第一坐标; 计算
每两个第一坐标之间的第一距离, 当第一距离小
于预设数值时, 判定行人之间的社交距离存在风
险并发出预警。 采用本发明实施例提高了对于存
在风险的社交距离的进行 预警的实时性。
权利要求书2页 说明书10页 附图1页
CN 115482277 A
2022.12.16
CN 115482277 A
1.一种社交距离风险预警方法, 其特 征在于, 包括:
将待检测图片输入至行人检测模型中, 以使所述行人检测模型根据所述待检测图片得
到多个行 人的边界框信息; 其中, 所述待检测图片中存在多个所述行 人;
根据所述 边界框信息, 结合BYTE跟踪算法, 计算得到多个行 人id信息;
根据所述行人id信息计算多个行人坐标, 对多个所述行人坐标依次进行逆透视变换,
分别生成多个第一 坐标;
计算每两个所述第一坐标之间的第一距离, 当所述第一距离小于预设数值时, 判定所
述行人之间的社交距离存在风险并发出 预警。
2.根据权利要求1所述的一种社交距离风险预警方法, 其特征在于, 所述行人检测模型
根据所述待检测图片得到多个行 人的边界框信息, 具体为:
将所述待检测图片输入至backbo ne模块, 输出 得到图片特 征;
将所述图片特 征依次输入至Neck模块和PANet模块, 输出 得到预测结果;
对所述预测结果进行非极大值抑制操作, 生成多个行人的边界框信 息及其对应的置信
度信息。
3.根据权利要求2所述的一种社交距离风险预警方法, 其特征在于, 所述行人检测模型
的训练过程具体为:
对训练数据集进行 预处理, 生成第一数据集;
对神经网络模型进行初始化操作后, 将所述第一数据集输入至所述神经网络模型中,
以使所述神经网络模 型输出得到特征列表; 其中, 所述特征列表包括: 边界框信息及其对应
的置信度信息和目标分类信息;
将所述特征列表和所述第 一数据集中的标签输入至损失函数计算损失, 并将所述损失
进行反向传播调整参数, 直至所述神经网络模型收敛, 停止训练并生成所述行 人检测模型。
4.根据权利要求3所述的一种社交距离风险预警方法, 其特征在于, 所述根据 所述边界
框信息, 结合BYTE跟踪算法, 计算得到多个行 人id信息, 具体为:
所述BYTE跟踪算法根据所述 边界框信息生成跟踪轨 迹;
根据所述边界框信 息所对应的置信度信 息, 将所述边界框信 息分成第 一类边界框信 息
和第二类边界框信息;
将所述第一类边界框信 息与跟踪轨迹进行匹配, 将匹配成功的跟踪轨迹标记为第 一跟
踪轨迹, 并生成第一匹配结果;
将所述第 二类边界框信 息与第二跟踪轨迹进行匹配, 生成第 二匹配结果; 其中, 所述第
二跟踪轨 迹为没有与所述第一类边界框信息匹配成功的跟踪轨 迹;
将没有匹配成功的所述第 一类边界框信 息与新建跟踪轨迹进行匹配, 生成第 三匹配结
果;
根据所述第一匹配结果、 所述第二匹配结果和所述第三匹配结果, 获得多个所述行人
id信息。
5.根据权利要求4所述的一种社交距离风险预警方法, 其特征在于, 所述计算每两个所
述第一坐标之间的第一距离, 具体为:
对所述待检测图片进行逆透 视变换, 生成鸟瞰图;
将多个所述第一 坐标标注至所述鸟瞰图中, 生成第一标注结果;权 利 要 求 书 1/2 页
2
CN 115482277 A
2从所述第一标注结果中计算每两个所述第一 坐标之间的第一距离 。
6.根据权利要求5所述的一种社交距离风险预警方法, 其特征在于, 所述当所述第 一距
离小于预设数值时, 判定所述行 人之间的社交距离存在风险并发出 预警, 具体为:
当所述第一距离小于预设数值时, 判定所述行人之间的社交距离存在风险, 并将所述
鸟瞰图转换回原 始格式, 并发出 预警。
7.根据权利要求6所述的一种社交距离风险预警方法, 其特征在于, 还包括: 获取所述
待检测图片, 具体为:
实时获取视频帧图像, 将所述视频帧图像裁剪为预设尺寸后, 生成所述待检测图片并
获取。
8.根据权利要求7 所述的一种社交距离风险预警方法, 其特 征在于, 还 包括:
使用TensorRT对所述行 人检测模型进行推理加速, 具体为:
将所述行 人检测模型从pytorc h模型转为通用的ON NX格式, 生成第一模型;
将所述第一模型转换为TensorRT模型, 进行反序列化, 创 建一个runtime engine并输
入检测数据, 以使所述TensorRT模型输出检测结果, 完成推理加速 。
9.一种社交距离风险预警装置, 其特征在于, 包括: 检测模块、 第一计算模块、 第 二计算
模块和预警模块;
其中, 所述检测模块用于将待检测图片输入至行人检测模型中, 以使所述行人检测模
型根据所述待检测图片得到多个行人 的边界框信息; 其中, 所述待检测图片 中存在多个所
述行人;
所述第一计算模块用于根据所述边界框信息, 结合BYTE跟踪算法, 计算得到多个行人
id信息;
所述第二计算模块用于根据 所述行人id信 息计算多个行人坐标, 对多个所述行人坐标
依次进行逆透 视变换, 分别生成多个第一 坐标;
所述预警模块用于计算每两个所述第 一坐标之间的第 一距离, 当所述第 一距离小于预
设数值时, 判定所述行 人之间的社交距离存在风险并发出 预警。
10.根据权利要求9所述的一种社交距离风险预警装置, 其特征在于, 所述行人检测模
型根据所述待检测图片得到多个行 人的边界框信息, 具体为:
将所述待检测图片输入至backbo ne模块, 输出 得到图片特 征;
将所述图片特 征依次输入至Neck模块和PANet模块, 输出 得到预测结果;
对所述预测结果进行非极大值抑制操作, 生成多个行人的边界框信 息及其对应的置信
度信息。权 利 要 求 书 2/2 页
3
CN 115482277 A
3
专利 一种社交距离风险预警方法及装置
文档预览
中文文档
14 页
50 下载
1000 浏览
0 评论
309 收藏
3.0分
温馨提示:本文档共14页,可预览 3 页,如浏览全部内容或当前文档出现乱码,可开通会员下载原始文档
本文档由 人生无常 于 2024-03-18 12:04:02上传分享