tests/nnapi/specs/skip/V1_2/generate_proposals.mod.py

   1 #
   2 # Copyright (C) 2019 The Android Open Source Project
   3 #
   4 # Licensed under the Apache License, Version 2.0 (the "License");
   5 # you may not use this file except in compliance with the License.
   6 # You may obtain a copy of the License at
   7 #
   8 #      http://www.apache.org/licenses/LICENSE-2.0
   9 #
  10 # Unless required by applicable law or agreed to in writing, software
  11 # distributed under the License is distributed on an "AS IS" BASIS,
  12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 # See the License for the specific language governing permissions and
  14 # limitations under the License.
  15 #
  16
  17 layout = BoolScalar("layout", False) # NHWC
  18
  19 # TEST 1: GENERATE_PROPOSALS_1
  20 model = Model()
  21 i1 = Input("scores", "TENSOR_FLOAT32", "{1, 2, 2, 2}") # scores
  22 i2 = Input("bboxDeltas", "TENSOR_FLOAT32", "{1, 2, 2, 8}") # bounding box deltas
  23 i3 = Input("anchors", "TENSOR_FLOAT32", "{2, 4}") # anchors
  24 i4 = Input("imageInfo", "TENSOR_FLOAT32", "{1, 2}") # image info
  25 o1 = Output("scoresOut", "TENSOR_FLOAT32", "{4}") # scores out
  26 o2 = Output("roiOut", "TENSOR_FLOAT32", "{4, 4}") # roi out
  27 o3 = Output("batchSplit", "TENSOR_INT32", "{4}") # batch split out
  28 model = model.Operation("GENERATE_PROPOSALS",
  29     i1, i2, i3, i4, 4.0, 4.0, -1, -1, 0.30, 1.0, layout).To(o1, o2, o3)
  30
  31 quant8 = DataTypeConverter().Identify({
  32     i1: ("TENSOR_QUANT8_ASYMM", 0.01, 100),
  33     i2: ("TENSOR_QUANT8_ASYMM", 0.05, 128),
  34     i3: ("TENSOR_QUANT16_SYMM", 0.125, 0),
  35     i4: ("TENSOR_QUANT16_ASYMM", 0.125, 0),
  36     o1: ("TENSOR_QUANT8_ASYMM", 0.01, 100),
  37     o2: ("TENSOR_QUANT16_ASYMM", 0.125, 0)
  38 })
  39
  40 input0 = {
  41     i1: [   # scores
  42         0.8, 0.9, 0.85, 0.85,
  43         0.75, 0.8, 0.9, 0.95
  44     ],
  45     i2: [   # bounding box deltas
  46         0.5, 0.1, 0.1, 0.1, 0.5, 0.1, 0.5, 0.1,
  47         -0.25, 0.1, -0.1, -0.1, -0.25, 0.1, 0.2, 0.1,
  48         0.4, -0.1, -0.2, 0.2, 0.4, -0.1, -0.2, 0.2,
  49         -0.2, -0.2, 0.2, 0.2, -0.2, -0.2, 0.2, 0.2
  50     ],
  51     i3: [0, 1, 4, 3, 1, 0, 3, 4],    # anchors
  52     i4: [32, 32],  # image info
  53 }
  54
  55 output0 = {
  56     o1: [0.95, 0.9, 0.85, 0.8],  # scores out
  57     o2: [   # roi out
  58         4.3785973,  2.7571943 , 6.8214025,  7.642805,
  59         1.3512788,  0.18965816, 4.648721 ,  4.610342,
  60         3.1903253,  1.2951627 , 6.8096747,  3.1048374,
  61         1.9812691,  3.1571944 , 3.6187308,  8.042806
  62     ],
  63     o3: [0, 0, 0, 0]
  64 }
  65
  66 Example((input0, output0)).AddNchw(i1, i2, layout).AddVariations("relaxed", quant8, "float16")
  67
  68 # TEST 2: GENERATE_PROPOSALS_2
  69 model = Model()
  70 i1 = Input("scores", "TENSOR_FLOAT32", "{2, 4, 4, 4}") # scores
  71 i2 = Input("bboxDeltas", "TENSOR_FLOAT32", "{2, 4, 4, 16}") # bounding box deltas
  72 i3 = Input("anchors", "TENSOR_FLOAT32", "{4, 4}") # anchors
  73 i4 = Input("imageInfo", "TENSOR_FLOAT32", "{2, 2}") # image info
  74 o1 = Output("scoresOut", "TENSOR_FLOAT32", "{30}") # scores out
  75 o2 = Output("roiOut", "TENSOR_FLOAT32", "{30, 4}") # roi out
  76 o3 = Output("batchSplit", "TENSOR_INT32", "{30}") # batch split out
  77 model = model.Operation("GENERATE_PROPOSALS",
  78     i1, i2, i3, i4, 10.0, 10.0, 32, 16, 0.20, 1.0, layout).To(o1, o2, o3)
  79
  80 quant8 = DataTypeConverter().Identify({
  81     i1: ("TENSOR_QUANT8_ASYMM", 0.005, 0),
  82     i2: ("TENSOR_QUANT8_ASYMM", 0.1, 128),
  83     i3: ("TENSOR_QUANT16_SYMM", 0.125, 0),
  84     i4: ("TENSOR_QUANT16_ASYMM", 0.125, 0),
  85     o1: ("TENSOR_QUANT8_ASYMM", 0.005, 0),
  86     o2: ("TENSOR_QUANT16_ASYMM", 0.125, 0)
  87 })
  88
  89 input0 = {
  90     i1: [   # scores
  91         0.885, 0.21 , 0.78 , 0.57 ,
  92         0.795, 0.66 , 0.915, 0.615,
  93         0.27 , 0.69 , 0.645, 0.945,
  94         0.465, 0.345, 0.855, 0.555,
  95         0.48 , 0.6  , 0.735, 0.63 ,
  96         0.495, 0.03 , 0.12 , 0.225,
  97         0.24 , 0.285, 0.51 , 0.315,
  98         0.435, 0.255, 0.585, 0.06 ,
  99         0.9  , 0.75 , 0.18 , 0.45 ,
 100         0.36 , 0.09 , 0.405, 0.15 ,
 101         0.   , 0.195, 0.075, 0.81 ,
 102         0.87 , 0.93 , 0.39 , 0.165,
 103         0.825, 0.525, 0.765, 0.105,
 104         0.54 , 0.705, 0.675, 0.3  ,
 105         0.42 , 0.045, 0.33 , 0.015,
 106         0.84 , 0.135, 0.72 , 0.375,
 107         0.495, 0.315, 0.195, 0.24 ,
 108         0.21 , 0.54 , 0.78 , 0.72 ,
 109         0.045, 0.93 , 0.27 , 0.735,
 110         0.135, 0.09 , 0.81 , 0.705,
 111         0.39 , 0.885, 0.42 , 0.945,
 112         0.9  , 0.225, 0.75 , 0.3  ,
 113         0.375, 0.63 , 0.825, 0.675,
 114         0.015, 0.48 , 0.645, 0.615,
 115         0.33 , 0.465, 0.66 , 0.6  ,
 116         0.075, 0.84 , 0.285, 0.57 ,
 117         0.585, 0.165, 0.06 , 0.36 ,
 118         0.795, 0.855, 0.105, 0.45 ,
 119         0.   , 0.87 , 0.525, 0.255,
 120         0.69 , 0.555, 0.15 , 0.345,
 121         0.03 , 0.915, 0.405, 0.435,
 122         0.765, 0.12 , 0.51 , 0.18
 123     ],
 124     i2: [   # bounding box deltas
 125     -1.9,  0.4,  1.4,  0.5, -1.5, -0.2,  0.3,  1.2,  0. , -0.6,  0.4, -1.3,  0.8,  0.9, -0.2,  0.8,
 126     -0.2,  0. ,  0.4,  0.1, -0.2, -1.6, -0.6, -0.1, -1. ,  0.6,  0.5, -0.2, -1.7, -1.4,  0.5, -0.1,
 127     -1.5,  1.3, -0.7, -0.9,  0.9,  0.2, -0.2,  0. , -0.7,  0.3, -0.4, -0.3, -0.5, -0.3,  1. , -0.7,
 128      1.2, -0.3,  0. ,  0.3, -0.7,  1. , -0.2, -0.6, -1.3,  0. ,  0.3,  0.1,  0.4,  0.2,  2.4,  0. ,
 129      0.1,  0. ,  0.7, -0.9,  0.1, -0.4,  0.3, -0.3, -0.7,  0.1,  0.7,  0. , -0.3,  1.6,  0. ,  1.1,
 130      0.4, -0.7, -0.9,  0. ,  0. ,  0.4, -0.6,  0.4, -1.9, -1.2,  0. , -0.3,  0.2,  0. ,  0.1,  0.8,
 131      0. ,  0.9, -1.7,  0.3,  0.7, -0.7,  0.7,  1.2, -0.4, -0.1, -0.6,  0.6, -0.4, -0.2,  0.3, -0.5,
 132      0. ,  1. , -0.1, -0.3, -0.8,  0.1, -1.2, -2.4,  0.1,  1.4,  0.4,  0.1, -1.1,  0.4, -0.4, -0.2,
 133      0.1,  0. ,  0.7,  0.1, -1.3,  0.1, -0.4, -0.2,  0.2,  0.1, -0.8,  0. , -1.4,  2. , -0.6, -0.5,
 134      0. ,  1. , -1.4, -1.1,  0.6, -0.7,  0.4,  1.1, -1.1,  1.6, -0.3,  0. , -0.7,  0.3, -1.3,  0. ,
 135      0. ,  0. , -0.3,  0. , -1.1, -1.5,  0.9, -1.4, -0.7,  0.1, -1.4,  0.9,  0.1,  0.2, -0.1, -1.7,
 136      0.2, -0.3, -0.9,  1.1,  0.1,  1. ,  1. , -0.9,  0.7,  0. , -0.3,  0.2, -0.8, -0.5,  0.6, -1.2,
 137      1. ,  0.6,  0. , -1.6,  0.1, -1.2,  0.7,  0.8,  0.5, -0.2, -0.8, -1.3, -0.3,  0. ,  0. ,  0.3,
 138     -0.6, -0.3,  1.3,  0.1,  2.2,  1.2, -1.1,  0.1,  1.2,  1.2,  1.3, -0.9,  0.1, -0.5,  0.1, -0.7,
 139     -1.3,  1.3,  0.1,  2. ,  0. ,  0.2,  0.6,  0. , -0.1, -0.4, -0.5,  0.1, -0.6, -0.3,  0.2, -0.4,
 140     -0.4, -0.7, -1.8,  0.4, -0.7,  0.4,  1.4, -0.3,  0.8,  0. ,  0.4, -0.1, -1. ,  0.2,  0.5, -0.6,
 141     -1.1,  0.2,  1.6, -0.2, -0.4, -0.9,  0. ,  0.3,  0. ,  0.3, -0.3,  0.3,  0.3,  1.9,  0.3, -0.5,
 142     -0.8, -1.3, -0.8,  0.2,  0.2, -0.4, -0.3,  0.6,  0.2, -0.2,  1.2,  0. ,  0. , -0.3,  0.3, -1.5,
 143     -1. , -0.3, -0.7, -0.3, -0.4, -1. , -0.6, -0.7, -0.2,  0.6, -0.3,  0.5, -0.2,  0.3, -0.5, -1.7,
 144      0. , -0.7, -0.1, -1.5, -0.9,  0.6,  0.3, -0.1,  0.2,  0.5,  0.6, -0.8, -0.3,  0.6,  0.9, -0.3,
 145      0.1, -1.7, -1.5,  0. , -0.1, -0.3,  0.7, -0.3, -0.4,  0. , -0.4, -0.3,  0.1,  1.1,  1.8, -0.9,
 146      0.6,  0.5,  0.2, -0.7,  0.2,  0.1,  1.2,  2.2,  0.3,  0.6,  0.4,  0.1,  0.2,  0. , -1.1, -0.2,
 147     -0.7,  0. , -1.2,  0.6, -0.6, -0.2, -0.4,  0. ,  0.7, -1.2,  0.8,  0. , -0.3,  0.2,  0.6, -1. ,
 148     -0.1, -0.1,  0. , -0.4, -0.2,  0.4, -1.4,  0.3,  0.1,  1.3, -0.2, -0.7,  0.6,  0.7,  0.6,  0.1,
 149     -0.4,  0.1, -0.2, -0.8,  0. , -1.3,  1.2,  1.4,  1.1,  0.5,  0.3,  0. ,  0.1, -0.4,  0.5, -0.1,
 150     -0.5,  0.3, -0.7,  0.9, -0.1, -0.4,  0.2, -0.8,  1. ,  1. ,  0.1,  0.1, -0.2,  0. , -0.4, -0.3,
 151     -0.8,  0.7, -0.9, -0.3, -0.3, -2.8,  1. ,  1.4,  0. , -2.6,  1.1, -1.1,  0.5,  0.1, -0.4, -1.5,
 152      0. ,  0.3, -0.3, -0.2,  0.7, -0.8, -0.1,  0.5,  0.7,  1.4, -1.2, -1. , -0.6,  0.2,  1.1, -0.9,
 153      0.7, -0.4,  0. ,  0. , -0.2, -0.2,  0.1,  0. ,  0. , -0.7, -0.7, -1.4, -0.9, -0.5, -0.6,  0.4,
 154      0.3,  0. ,  0.9, -0.2,  0.7,  1.2,  0.5,  0.8, -0.5,  1. ,  0.2, -0.5,  1.3, -0.5,  0.3,  1.2,
 155     -0.3, -0.1,  1.3,  0.2,  0.6, -1.4, -0.1, -0.2, -0.4, -0.9,  1.2, -0.9, -0.2, -1.2, -1. , -0.2,
 156     -1.6,  2.1, -0.6, -0.2, -0.3,  0.5,  0.9, -0.4,  0. , -0.1,  0.1, -0.6, -1. , -0.7,  0.2, -0.2
 157     ],
 158     i3: [    # anchors
 159         0, 6, 16, 10,
 160         6, 0, 10, 16,
 161         3, 5, 13, 11,
 162         5, 3, 11, 13
 163     ],
 164     i4: [64, 64, 32, 32],  # image info
 165 }
 166
 167 output0 = {
 168     o1: [  # scores out
 169         0.945, 0.93 , 0.915, 0.9  , 0.87 , 0.84 , 0.81, 0.795, 0.78, 0.765, 0.75, 0.735,
 170         0.72 , 0.705, 0.69 , 0.675, 0.945, 0.915, 0.9 , 0.885, 0.87, 0.84 , 0.81, 0.78,
 171         0.735, 0.72 , 0.63 , 0.6  , 0.585, 0.54
 172     ],
 173     o2: [   # roi out
 174         16.845154 ,  2.5170734, 33.154846 ,  7.4829264,
 175         32.96344  , 40.747444 , 43.836563 , 47.252556 ,
 176          0.       ,  9.143808 , 16.243607 , 14.056192 ,
 177          0.       , 25.789658 , 25.710022 , 30.210342 ,
 178         37.947445 , 20.791668 , 44.452557 , 32.80833  ,
 179         30.277609 , 32.21635  , 32.92239  , 38.18365  ,
 180         25.885489 , 29.086582 , 31.314512 , 30.913418 ,
 181          2.8654022,  5.789658 , 26.734598 , 10.210342 ,
 182          0.5408764,  3.5824041, 15.459124 ,  5.217595 ,
 183         10.753355 , 35.982403 , 15.246645 , 37.617596 ,
 184          1.4593601, 23.050154 ,  4.1406403, 36.149845 ,
 185          0.       , 15.6      , 11.068764 , 21.6      ,
 186         38.54088  , 35.28549  , 53.45912  , 40.71451  ,
 187         26.134256 , 48.358635 , 27.465742 , 64.       ,
 188         29.96254  ,  3.1999998, 33.23746  , 19.2      ,
 189         11.653517 , 43.980293 , 48.34648  , 46.41971  ,
 190          0.       , 26.967152 , 26.748941 , 31.032848 ,
 191         28.590324 ,  9.050154 , 32.       , 22.149847 ,
 192         17.828777 , 19.00683  , 32.       , 20.99317  ,
 193          3.5724945,  7.273454 , 11.627505 , 19.126545 ,
 194          4.989658 , 26.8      ,  9.410341 , 32.       ,
 195         15.157195 , 18.00537  , 20.042807 , 25.194632 ,
 196         30.889404 ,  9.652013 , 32.       , 12.347987 ,
 197          3.399414 ,  3.8000002, 32.       ,  9.8      ,
 198         24.980408 , 10.086582 , 28.61959  , 11.913418 ,
 199         13.950423 ,  3.884349 , 22.049576 ,  6.115651 ,
 200         24.259361 ,  6.8      , 26.94064  , 22.8      ,
 201          3.6538367, 19.475813 , 13.546164 , 28.524187 ,
 202         11.947443 , 29.318363 , 18.452557 , 32.       ,
 203         17.318363 ,  0.       , 20.281635 , 16.17695
 204     ],
 205     o3: [
 206         0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
 207         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1
 208     ]
 209 }
 210
 211 Example((input0, output0)).AddNchw(i1, i2, layout).AddVariations("relaxed", quant8, "float16")