昇腾社区首页
中文
注册
适配200DK的安全帽识别

概述

本应用以昇腾Atlas 200I DK A2为主要的硬件平台,使用MindX SDK中实现识别每个人是否戴上安全帽,并对没戴安全帽的人做出告警的功能

功能模块介绍

本案例流程如下:

1. 视频预处理:对输入的两路视频数据解码为YUV格式图片,将解码后的图片缩放为模型指定输入的大小
2. 安全帽模型推理:YOLOv5进行安全帽识别。识别两类:安全帽与人体头部识别,主要识别出没有佩戴安全帽的情况
3. 跟踪去重:对重复检测出的没戴安全帽的对象进行去重
4. 结果输出&告警:将识别结果输出,并对没佩戴安全帽的情况告警

支持说明

本应用同样支持Atlas200 DK,应用链接