commit 325279580fdadc36f82eb0aff2a2f3d72d5b71aa
author Manuel Bottini <manuel.bottini@arm.com> Tue Nov 05 16:55:57 2019 +0000
committer Georgios Pinitas <georgios.pinitas@arm.com> Wed Nov 06 11:25:05 2019 +0000
tree 81d690f9891d998ac22f71f61d704267b167bcbb
parent 0fc2d7ad56b297852375b34322d0920193a21b8b

COMPMID-2106: Extend get_input_accessor() to run input images through QASYMM8 networks

Change-Id: I5c44bcc95990c343d7829d885b93e8ff16691c50
Signed-off-by: Manuel Bottini <manuel.bottini@arm.com>
Reviewed-on: https://review.mlplatform.org/c/2226
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>

utils/ImageLoader.h

diff --git a/utils/ImageLoader.h b/utils/ImageLoader.h
index c03cbdf..497320e 100644
--- a/utils/ImageLoader.h
+++ b/utils/ImageLoader.h
@@ -257,7 +257,7 @@
     void fill_planar_tensor(T &tensor, bool bgr = false)
     {
         ARM_COMPUTE_ERROR_ON(!is_open());
-        ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(&tensor, 1, DataType::U8, DataType::F32, DataType::F16);
+        ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(&tensor, 1, DataType::U8, DataType::QASYMM8, DataType::F32, DataType::F16);
 
         const DataLayout  data_layout  = tensor.info()->data_layout();
         const TensorShape tensor_shape = tensor.info()->tensor_shape();
@@ -312,6 +312,7 @@
                 switch(tensor.info()->data_type())
                 {
                     case DataType::U8:
+                    case DataType::QASYMM8:
                     {
                         *(out.ptr() + 0 * stride_z) = bgr ? blue : red;
                         *(out.ptr() + 1 * stride_z) = green;