File size: 3,059 Bytes
477da44
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
# RKNN C API Dynamic Shape Input Demo

This is a demo that uses the RKNN C API for dynamic shape input inference. In this demo, you can see how to use the RKNN dynamic shape C API to perform image classification.

## How to Use

1. Clone or download this code repository: https://github.com/rockchip-linux/rknn-toolkit2/tree/master/rknpu2.
2. Navigate to the dynamic shape inference demo directory in your terminal.

```shell
cd examples/rknn_dynamic_shape_input_demo
```

3. Compile the application by running the shell script based on the chip platform. For example, for the RK3562 Android system, run the following command:

```shell
./build-android_RK3562.sh
```

4. Push the demo program directory to the target board's system using the adb command. For example:

```shell
#If using Android system, make sure to run adb root & adb remount first.
adb push ./install/rknn_dynshape_demo_Android/ /data
```

5. Set the runtime library path.

```
export LD_LIBRARY_PATH=./lib
```

6. Run the program. For example, on the RK3562 platform, use the command

   ```shell
   ./rknn_dynshape_inference model/RK3562/mobilenet_v2.rknn images/dog_224x224.jpg
   ```

   ,where `mobilenet_v2.rknn` is the name of the neural network model file, and `dog_224x224.jpg` is the name of the image file to classify.

## Compilation Instructions

### Arm Linux

First export `GCC_COMPILER`, for example `export GCC_COMPILER=~/opt/gcc-linaro-7.5.0-2019.12-x86_64_aarch64-linux-gnu/bin/aarch64-linux-gnu`, then execute:

```
./build-linux.sh -t <target> -a <arch> -b <build_type>]

# such as: 
./build-linux.sh -t rk3588 -a aarch64 -b Release
```

### Android

First export `ANDROID_NDK_PATH`, for example `export ANDROID_NDK_PATH=~/opts/ndk/android-ndk-r18b`, then execute:

```
./build-android.sh -t <target> -a <arch> [-b <build_type>]

# sush as: 
./build-android.sh -t rk3568 -a arm64-v8a -b Release
```

## Included Features

This demonstration application includes the following features:

- Creating a neural network model with dynamic shape inputs. Please refer to the examples/functions/dynamic_input directory in the https://github.com/rockchip-linux/rknn-toolkit2 repository for more information.
- Reading an image from a file and performing classification using the neural network model. The program follows these steps:

1. Initialize the RKNN context using the `rknn_init()` function.
2. Set the shape information of all the model inputs using the `rknn_set_input_shapes()` function, including shape and layout.
3. Query the current model input and output information, including shape, data type, and size, using the `rknn_query()` function.
4. Set the input data of the model using the `rknn_inputs_set()` function, including data pointer and size.
5. Run the model using the `rknn_run()` function.
6. Retrieve the output data by using the `rknn_outputs_get()` function, specifying the need for float-type results.
7. Process the output data to obtain the classification results and probabilities.
8. Release the RKNN context using the `rknn_release()` function.