AIAS/1_image_sdks/security/smart_construction_sdk
2021-12-12 23:09:15 +08:00
..
build/output no message 2021-11-28 17:35:30 +08:00
src no message 2021-11-28 17:35:30 +08:00
pom.xml Fix Log4j2 漏洞 2021-12-12 23:09:15 +08:00
README.md no message 2021-10-29 10:29:42 +08:00
smart_construction_sdk.iml no message 2021-11-28 17:35:30 +08:00

智慧工地检测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]
]

帮助

引擎定制化配置,可以提升首次运行的引擎下载速度,解决外网无法访问或者带宽过低的问题。
引擎定制化配置

官网:

官网链接

Git地址

Github链接
Gitee链接