diff options
author | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-21 17:10:53 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-21 17:10:53 -0700 |
commit | 501a1955cf44568641f7205c27852eed20bc62db (patch) | |
tree | dfa074fb5eb6e4781b747df9c5439014ee48078c /tensorflow/contrib/lite/examples | |
parent | 305a392904e6981e935c2a3514394379ba7083b1 (diff) | |
parent | 9fd4798cc2a4b1cc20d5577f944c6423a8aaabef (diff) |
Merge pull request #22334 from freedomtan:tflite_label_image_output_pixels
PiperOrigin-RevId: 214067431
Diffstat (limited to 'tensorflow/contrib/lite/examples')
-rw-r--r-- | tensorflow/contrib/lite/examples/label_image/bitmap_helpers_impl.h | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/examples/label_image/bitmap_helpers_impl.h b/tensorflow/contrib/lite/examples/label_image/bitmap_helpers_impl.h index 6fdcf78b69..21ad39a6bf 100644 --- a/tensorflow/contrib/lite/examples/label_image/bitmap_helpers_impl.h +++ b/tensorflow/contrib/lite/examples/label_image/bitmap_helpers_impl.h @@ -80,8 +80,7 @@ void resize(T* out, uint8_t* in, int image_height, int image_width, interpreter->Invoke(); auto output = interpreter->typed_tensor<float>(2); - auto output_number_of_pixels = - wanted_height * wanted_height * wanted_channels; + auto output_number_of_pixels = wanted_height * wanted_width * wanted_channels; for (int i = 0; i < output_number_of_pixels; i++) { if (s->input_floating) |