智慧工地检测SDK

支持检测的类别:

  • person (人体)
  • head (没戴安全帽)
  • helmet (戴安全帽)

SDK功能

工地安全检测,给出检测框和置信度。

  • 提供三个模型:
  • 小模型(yolov5s 29.7M)
  • 中模型(yolov5m 86.8M)
  • 大模型(yolov5l 190.8M)

运行小模型例子 - Yolov5sExample

  • 测试图片效果(只显示安全帽检测,过滤了其它类别的显示,具体看代码)
    small

运行中模型例子 - Yolov5mExample

  • 测试图片效果
    medium

运行大模型例子 - Yolov5lExample

  • 测试图片效果
    large

运行成功后,命令行应该看到下面的信息:

[INFO ] - [
	class: "helmet", probability: 0.89502, bounds: [x=0.956, y=0.525, width=0.044, height=0.067]
	class: "helmet", probability: 0.85951, bounds: [x=0.237, y=0.439, width=0.036, height=0.046]
	class: "helmet", probability: 0.81705, bounds: [x=0.901, y=0.378, width=0.036, height=0.052]
	class: "helmet", probability: 0.80817, bounds: [x=0.250, y=0.399, width=0.029, height=0.040]
	class: "helmet", probability: 0.80528, bounds: [x=0.771, y=0.336, width=0.029, height=0.043]
]

目录:

http://www.aias.top/

Git地址:

https://github.com/mymagicpower/AIAS
https://gitee.com/mymagicpower/AIAS

Logo

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐