1.9 KiB

Raw Blame History

How to use dynamic shape function

Model Source

The model used in this example come from the following open source projects:
https://github.com/shicai/MobileNet-Caffe

Convert to RKNN model

Please refer to the example in the RKNN Toolkit2 project to generate the RKNN model: https://github.com/rockchip-linux/rknn-toolkit2/tree/master/examples/functions/dynamic_shape

Script Usage

Usage:

python test.py

Expected Results

This example will print the TOP5 labels and corresponding scores of the test image classification results for each different input shape, as follows:

model: mobilenet_v2

input shape: 1,3,224,224
W The input[0] need NHWC data format, but NCHW set, the data format and data buffer will be changed to NHWC.
-----TOP 5-----
[155] score:0.992188 class:"Shih-Tzu"
[154] score:0.002636 class:"Pekinese, Pekingese, Peke"
[204] score:0.002636 class:"Lhasa, Lhasa apso"
[283] score:0.001698 class:"Persian cat"
[896] score:0.000273 class:"washbasin, handbasin, washbowl, lavabo, wash-hand basin"

input shape: 1,3,160,160
W The input[0] need NHWC data format, but NCHW set, the data format and data buffer will be changed to NHWC.
-----TOP 5-----
[155] score:0.558594 class:"Shih-Tzu"
[154] score:0.408447 class:"Pekinese, Pekingese, Peke"
[204] score:0.031036 class:"Lhasa, Lhasa apso"
[194] score:0.000956 class:"Dandie Dinmont, Dandie Dinmont terrier"
[219] score:0.000256 class:"cocker spaniel, English cocker spaniel, cocker"

input shape: 1,3,256,256
W The input[0] need NHWC data format, but NCHW set, the data format and data buffer will be changed to NHWC.
-----TOP 5-----
[155] score:0.980957 class:"Shih-Tzu"
[154] score:0.008835 class:"Pekinese, Pekingese, Peke"
[204] score:0.004883 class:"Lhasa, Lhasa apso"
[193] score:0.000929 class:"Australian terrier"
[200] score:0.000509 class:"Tibetan terrier, chrysanthemum dog"

Note: Different platforms, different versions of tools and drivers may have slightly different results.

1.9 KiB Raw Blame History