| # How to expand batch for use multi-batch function | |
| ## Model Source | |
| The model used in this example come from the following open source projects: | |
| https://github.com/shicai/MobileNet-Caffe | |
| ## Script Usage | |
| *Usage:* | |
| ``` | |
| python test.py | |
| ``` | |
| *Description:* | |
| - The default target platform in script is 'rk3566', please modify the 'target_platform' parameter of 'rknn.config' according to the actual platform. | |
| - If connecting board is required, pl'ease add the 'target' parameter in 'rknn.init_runtime'. | |
| - You can modify the 'rknn_batch_size' parameter of 'rknn.build' to achieve the effect of multi-batch. | |
| ## Expected Results | |
| This example will print the TOP5 labels and corresponding scores of the test image classification results for every batch, as follows: | |
| ``` | |
| ----- Batch 0: TOP 5 ----- | |
| [155] score:0.993652 class:"Shih-Tzu" | |
| [154] score:0.002834 class:"Pekinese, Pekingese, Peke" | |
| [204] score:0.002172 class:"Lhasa, Lhasa apso" | |
| [283] score:0.000571 class:"Persian cat" | |
| [284] score:0.000133 class:"Siamese cat, Siamese" | |
| ----- Batch 1: TOP 5 ----- | |
| [ 1] score:0.598145 class:"goldfish, Carassius auratus" | |
| [794] score:0.062073 class:"shower curtain" | |
| [996] score:0.062073 class:"hen-of-the-woods, hen of the woods, Polyporus frondosus, Grifola frondosa" | |
| [927] score:0.041687 class:"trifle" | |
| [115] score:0.027863 class:"sea slug, nudibranch" | |
| ----- Batch 2: TOP 5 ----- | |
| [812] score:0.999023 class:"space shuttle" | |
| [148] score:0.000293 class:"killer whale, killer, orca, grampus, sea wolf, Orcinus orca" | |
| [517] score:0.000132 class:"crane" | |
| [833] score:0.000132 class:"submarine, pigboat, sub, U-boat" | |
| [525] score:0.000089 class:"dam, dike, dyke" | |
| ----- Batch 3: TOP 5 ----- | |
| [155] score:0.993652 class:"Shih-Tzu" | |
| [154] score:0.002834 class:"Pekinese, Pekingese, Peke" | |
| [204] score:0.002172 class:"Lhasa, Lhasa apso" | |
| [283] score:0.000571 class:"Persian cat" | |
| [284] score:0.000133 class:"Siamese cat, Siamese" | |
| ``` | |
| - Note: Different platforms, different versions of tools and drivers may have slightly different results. |