SW: imx-yocto-5.4.70-2.3.0
HW: i.MX8M Plus
We built an image segmentation model using onnx.
The average inference time is 2.77394s on CPU per frame. The performance drops to 21.6163s if VSI NPU EP is used instead.
Attached the files:
- main.cpp: inference on CPU
- main_npu.cpp: inference on NPU
- video.mp4: test video
- model.onnx: image segmentation model
Just build the code and run it. It reads video file and performs segmentation.
As for enabling NPU, it is to add VSI NPU EP which list in IMX-MACHINE-LEARNING-UG.pdf. You can refer to my main_npu.cpp.