shen/AIAS1
1
0
Fork 0
mirror of https://gitee.com/mymagicpower/AIAS.git synced 2024-11-25 16:05:33 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
main
AIAS1/1_image_sdks/safety_helmet_sdk
History
Calvin 7b21cc0960 no message
2024-11-25 10:43:10 +08:00
..
models no message 2024-11-25 10:43:10 +08:00
src no message 2024-11-25 10:43:10 +08:00
pom.xml no message 2024-11-25 10:43:10 +08:00
README_CN.md no message 2024-11-25 10:43:10 +08:00
README_EN.md no message 2024-11-25 10:43:10 +08:00
safety_helmet_sdk.iml no message 2024-11-25 10:43:10 +08:00
safety-helmet-sdk.iml no message 2024-11-25 10:43:10 +08:00

README_EN.md

Download the model, place it in the /models directory, and unzip

Safety Helmet Detection SDK

Safety helmet detection. -Support categories: -Safe -Unsafe

SDK functions

  • Safety helmet detection, giving detection boxes and confidence. -Three models: -Small model (mobilenet0.25) -Medium model (mobilenet1.0) -Large model (darknet53)

Run the small model example - SmallSafetyHelmetDetectExample

  • Test picture small

Run the medium model example - MediumSafetyHelmetDetectExample

  • Test picture medium

Run the large model example - LargeSafetyHelmetDetectExample

  • Test picture large

After successful operation, the command line should see the following information:

[INFO ] - [
	class: "safe 0.9983590245246887", probability: 0.99835, bounds: [x=0.244, y=0.000, width=0.086, height=0.150]
	class: "unsafe 0.998088538646698", probability: 0.99808, bounds: [x=0.226, y=0.204, width=0.115, height=0.263]
	class: "safe 0.997364342212677", probability: 0.99736, bounds: [x=0.584, y=0.247, width=0.162, height=0.302]
	class: "safe 0.9963852167129517", probability: 0.99638, bounds: [x=0.319, y=0.000, width=0.076, height=0.133]
	class: "safe 0.9952006936073303", probability: 0.99520, bounds: [x=0.757, y=0.262, width=0.111, height=0.264]
]

Open source algorithm

Open source algorithm used by the SDK