DocumentAIKit/deploy/third_engine/demo_avh

Running PP-PicoDet object detection model on bare metal Arm(R) Cortex(R)-M55 CPU using Arm Virtual Hardware

This folder contains an example of how to run a PP-PicoDet model on bare metal Cortex(R)-M55 CPU using Arm Virtual Hardware.

Running environment and prerequisites

Case 1: If the demo is run in Arm Virtual Hardware Amazon Machine Image(AMI) instance hosted by AWS/AWS China, the following software will be installed through configure_avh.sh script. It will install automatically when you run the application through run_demo.sh script. You can refer to this guide to launch an Arm Virtual Hardware AMI instance.

Case 2: If the demo is run in the ci_cpu Docker container provided with TVM, then the following software will already be installed.

Case 3: If the demo is not run in the ci_cpu Docker container, then you will need the following:

• Software required to build and run the demo (These can all be installed by running tvm/docker/install/ubuntu_install_ethosu_driver_stack.sh.)
  • Fixed Virtual Platform (FVP) based on Arm(R) Corstone(TM)-300 software
  • cmake 3.19.5
  • GCC toolchain from Arm(R)
  • Arm(R) Ethos(TM)-U NPU driver stack
  • CMSIS
• The python libraries listed in the requirements.txt of this directory
  • These can be installed by running the following from the current directory: bash pip install -r ./requirements.txt

In case2 and case3:

You will need to update your PATH environment variable to include the path to cmake 3.19.5 and the FVP. For example if you've installed these in /opt/arm , then you would do the following:

export PATH=/opt/arm/FVP_Corstone_SSE-300/models/Linux64_GCC-6.4:/opt/arm/cmake/bin:$PATH

You will also need TVM which can either be:

• Installed from TLCPack(see TLCPack)
• Built from source (see Install from Source)
  • When building from source, the following need to be set in config.cmake:
    • set(USE_CMSISNN ON)
    • set(USE_MICRO ON)
    • set(USE_LLVM ON)

Running the demo application

Type the following command to run the bare metal text recognition application (src/demo_bare_metal.c):

./run_demo.sh

If you are not able to use Arm Virtual Hardware Amazon Machine Image(AMI) instance hosted by AWS/AWS China, specify argument --enable_FVP to 1 to make the application run on local Fixed Virtual Platforms (FVPs) executables.

./run_demo.sh --enable_FVP 1

If the Ethos(TM)-U platform and/or CMSIS have not been installed in /opt/arm/ethosu then the locations for these can be specified as arguments to run_demo.sh, for example:

./run_demo.sh --cmsis_path /home/tvm-user/cmsis \
--ethosu_platform_path /home/tvm-user/ethosu/core_platform

With run_demo.sh to run the demo application, it will:

• Set up running environment by installing the required prerequisites automatically if running in Arm Virtual Hardware Amazon AMI instance(not specify --enable_FVP to 1)
• Download a PP-PicoDet model
• Use tvmc to compile the text recognition model for Cortex(R)-M55 CPU and CMSIS-NN
• Create a C header file inputs.c containing the image data as a C array
• Create a C header file outputs.c containing a C array where the output of inference will be stored
• Build the demo application
• Run the demo application on a Arm Virtual Hardware based on Arm(R) Corstone(TM)-300 software
• The application will report the text on the image and the corresponding score.

Using your own image

The create_image.py script takes a single argument on the command line which is the path of the image to be converted into an array of bytes for consumption by the model.

The demo can be modified to use an image of your choice by changing the following line in run_demo.sh

python3 ./convert_image.py path/to/image

Model description

In this demo, the model we used is based on PP-PicoDet. Because of the excellent performance, PP-PicoDet are very suitable for deployment on mobile or CPU. And it is released by PaddleDetection.