diff options
| author |  Koan-Sin Tan <koansin.tan@gmail.com> | 2018-09-18 10:40:54 +0800 |
|---|---|---|
| committer | Koan-Sin Tan <koansin.tan@gmail.com> | 2018-09-18 10:40:54 +0800 |
| commit | 2cb119b81fd08a1e680a2b44ff68c0a8c76eb017 (patch) | |
| tree | 907c53c177fc77fc667c7fdc9e9b1cfde67386c9 /tensorflow/contrib/lite/examples | |
| parent | 71fab28dc4741dedf13fea732f6b134608719bc7 (diff) | |
[tflite] fix calculating of output pixels
fix an issue reported by issue #22310
Diffstat (limited to 'tensorflow/contrib/lite/examples')
| -rw-r--r-- | tensorflow/contrib/lite/examples/label_image/bitmap_helpers_impl.h | 2 |
1 files changed, 1 insertions, 1 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..7e09d4bc79 100644 --- a/tensorflow/contrib/lite/examples/label_image/bitmap_helpers_impl.h +++ b/tensorflow/contrib/lite/examples/label_image/bitmap_helpers_impl.h @@ -81,7 +81,7 @@ void resize(T* out, uint8_t* in, int image_height, int image_width, auto output = interpreter->typed_tensor<float>(2); auto output_number_of_pixels = - wanted_height * wanted_height * wanted_channels; + wanted_height * wanted_width * wanted_channels; for (int i = 0; i < output_number_of_pixels; i++) { if (s->input_floating) |