AXERA-TECH/Person_car-axera

Axera-PCD

This version of Axera-PCD has been converted to run on the Axera NPU using w8a16 quantization. It is mainly used for person and car detection in conventional scenarios.

Supported Classes

This model is trained to detect the following 3 classes:

1. person
2. car
3. person_cycle

Compatible with Pulsar2 version: 5.0.

Convert tools links:

For those who are interested in model conversion, you can try to export axmodel through:

• The repo of AXera Platform, where you can get the detailed guide.
• Pulsar2 Link, How to Convert ONNX to axmodel

Support Platform

https://docs.m5stack.com/zh_CN/ai_hardware/AI_Pyramid-Pro

• AX650N/AX8850
  • M4N-Dock(爱芯派Pro)
  • AI Pyramid
  • M.2 Accelerator card
• AX620E
  • 爱芯派2
  • Module-LLM

How to use

Download all files from this repository to the device.

python env requirement

pyaxengine

https://github.com/AXERA-TECH/pyaxengine

wget https://github.com/AXERA-TECH/pyaxengine/releases/download/0.1.3.rc2/axengine-0.1.3-py3-none-any.whl
pip install axengine-0.1.3-py3-none-any.whl

Inference with AX650 Host, such as M4N-Dock(爱芯派Pro)

Input image:

run

python3 ax_pcd_infer.py --model ./ax_ax650_pcd_algo_V2.0.0.axmodel --img car_away_1920x1080.jpg

root@ax650:/pcd# python3 ax_pcd_infer.py  --model ./ax_ax650_pcd_algo_V2.0.0.axmodel --img car_away_1920x1080.jpg
[INFO] Available providers:  ['AxEngineExecutionProvider']
[INFO] Using provider: AxEngineExecutionProvider
[INFO] Chip type: ChipType.MC50
[INFO] VNPU type: VNPUType.DISABLED
[INFO] Engine version: 2.10.1s
[INFO] Model type: 2 (triple core)
[INFO] Compiler version: 6.0 93b95f7f
Input_name: images, Output_name: ['/model.20/Concat_output_0', '/model.20/Concat_1_output_0']
Preprocess time: 7.83 ms
Inference time: 20.87 ms
Postprocess time: 5.97 ms
Total detect 14 objects
0: person        0.824 [1727.0, 768.0, 1795.0, 925.0]
1: person        0.674 [1315.0, 617.0, 1356.0, 726.0]
2: car   0.911 [379.0, 665.0, 533.0, 801.0]
3: car   0.911 [662.0, 811.0, 860.0, 975.0]
4: car   0.911 [427.0, 812.0, 643.0, 1050.0]
5: car   0.911 [32.0, 919.0, 277.0, 1080.0]
6: car   0.824 [449.0, 532.0, 565.0, 619.0]
7: car   0.824 [111.0, 676.0, 266.0, 811.0]
8: car   0.674 [123.0, 336.0, 168.0, 367.0]
9: car   0.674 [267.0, 404.0, 328.0, 454.0]
10: car  0.674 [182.0, 434.0, 252.0, 486.0]
11: car  0.674 [330.0, 555.0, 451.0, 657.0]
12: car  0.465 [451.0, 402.0, 508.0, 447.0]
13: car  0.465 [497.0, 493.0, 597.0, 566.0]
Result saved to ./out.jpg

Output image: