Package-axera

Axera-Package

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

Supported Classes

This model is trained to detect the following 2 classes:

1. package
2. package-bag

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_pkg_infer.py --model ./AX650/ax_ax650_package_algo_model_V2.0.0.axmodel --img package.jpg

root@ax650:~/pkg# python3 ax_pkg_infer.py --model ./AX650/ax_ax650_package_algo_model_V2.0.0.axmodel --img package.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 79a1e641
Input_name: images,      Output_name: ['output0', 'output1']
Preprocess time: 7.43 ms
Inference time: 17.77 ms
Total detect 1 objects
0: package       0.915 [603.0, 258.0, 722.0, 446.0]

Output image: