安检作为在公共场合保障群众人身安全的重要环节,在各个场合中都必不可少,x光安检机作为安检的重要工具,其人工监控判断成像的特点使其具有很大的局限性。本文以此为出发点,研究了Torch框架下YOLO
实现了对安检仪x光图像中刀具、气罐、鞭炮等10类危险物品的目标检测功能,为YOLOv5算法搭建了界面,实现了图像识别,视频时别和实时摄像识别。

源码和数据集见视频简介:

【源码解析】基于yolov5的安检仪危险品识别
https://www.bilibili.com/video/BV1VG4y1H7Ns/

1、检测效果:

在这里插入图片描述

2、数据集

包含了1288张yolo格式的数据集。

3、训练过程

在这里插入图片描述

4、训练结果

请添加图片描述

5、总结

欢迎互相讨论学习,评论看到就会回。

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐