概述
本应用以昇腾Atlas 500 A2、Atlas 300I Pro、Atlas 300V Pro为主要的硬件平台,基于Vision SDK实现识别每个人是否戴上安全帽,并对没戴安全帽的人做出告警的功能
功能模块介绍
本案例流程如下:
1. 视频预处理:对输入的两路视频数据解码为YUV格式图片,将解码后的图片缩放为模型指定输入的大小
2. 安全帽模型推理:YOLOv5进行安全帽识别。识别两类:安全帽与人体头部识别,主要识别出没有佩戴安全帽的情况
3. 跟踪去重:对重复检测出的没戴安全帽的对象进行去重
4. 结果输出&告警:将识别结果输出,并对没佩戴安全帽的情况告警
效果展示
暂无