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