diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/use_case/img_class/InferenceTestMobilenetV2.cc | 4 | ||||
-rw-r--r-- | tests/use_case/object_detection/InferenceTestYoloFastest.cc | 4 | ||||
-rw-r--r-- | tests/use_case/vww/InferenceVisualWakeWordModelTests.cc | 4 | ||||
-rw-r--r-- | tests/utils/ImageUtils.cc | 37 | ||||
-rw-r--r-- | tests/utils/ImageUtils.hpp | 26 |
5 files changed, 6 insertions, 69 deletions
diff --git a/tests/use_case/img_class/InferenceTestMobilenetV2.cc b/tests/use_case/img_class/InferenceTestMobilenetV2.cc index 294215f..7e7508b 100644 --- a/tests/use_case/img_class/InferenceTestMobilenetV2.cc +++ b/tests/use_case/img_class/InferenceTestMobilenetV2.cc @@ -1,5 +1,5 @@ /* - * Copyright (c) 2021 Arm Limited. All rights reserved. + * Copyright (c) 2021-2022 Arm Limited. All rights reserved. * SPDX-License-Identifier: Apache-2.0 * * Licensed under the Apache License, Version 2.0 (the "License"); @@ -34,7 +34,7 @@ bool RunInference(arm::app::Model& model, const int8_t imageData[]) memcpy(inputTensor->data.data, imageData, copySz); if(model.IsDataSigned()){ - convertImgIoInt8(inputTensor->data.data, copySz); + arm::app::image::ConvertImgToInt8(inputTensor->data.data, copySz); } return model.RunInference(); diff --git a/tests/use_case/object_detection/InferenceTestYoloFastest.cc b/tests/use_case/object_detection/InferenceTestYoloFastest.cc index d9cabbd..b3bd408 100644 --- a/tests/use_case/object_detection/InferenceTestYoloFastest.cc +++ b/tests/use_case/object_detection/InferenceTestYoloFastest.cc @@ -66,10 +66,10 @@ bool RunInference(arm::app::Model& model, const uint8_t imageData[]) const size_t copySz = inputTensor->bytes < IMAGE_DATA_SIZE ? inputTensor->bytes : IMAGE_DATA_SIZE; - image::RgbToGrayscale(imageData,inputTensor->data.uint8,copySz); + arm::app::image::RgbToGrayscale(imageData,inputTensor->data.uint8,copySz); if(model.IsDataSigned()){ - convertImgIoInt8(inputTensor->data.data, copySz); + arm::app::image::ConvertImgToInt8(inputTensor->data.data, copySz); } return model.RunInference(); diff --git a/tests/use_case/vww/InferenceVisualWakeWordModelTests.cc b/tests/use_case/vww/InferenceVisualWakeWordModelTests.cc index 04dce0d..194099f 100644 --- a/tests/use_case/vww/InferenceVisualWakeWordModelTests.cc +++ b/tests/use_case/vww/InferenceVisualWakeWordModelTests.cc @@ -1,5 +1,5 @@ /* - * Copyright (c) 2021 Arm Limited. All rights reserved. + * Copyright (c) 2021-2022 Arm Limited. All rights reserved. * SPDX-License-Identifier: Apache-2.0 * * Licensed under the Apache License, Version 2.0 (the "License"); @@ -34,7 +34,7 @@ bool RunInference(arm::app::Model& model, const int8_t* imageData) memcpy(inputTensor->data.data, imageData, copySz); if(model.IsDataSigned()){ - convertImgIoInt8(inputTensor->data.data, copySz); + arm::app::image::ConvertImgToInt8(inputTensor->data.data, copySz); } return model.RunInference(); diff --git a/tests/utils/ImageUtils.cc b/tests/utils/ImageUtils.cc deleted file mode 100644 index 506040f..0000000 --- a/tests/utils/ImageUtils.cc +++ /dev/null @@ -1,37 +0,0 @@ -/* - * Copyright (c) 2021 Arm Limited. All rights reserved. - * SPDX-License-Identifier: Apache-2.0 - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - * See the License for the specific language governing permissions and - * limitations under the License. - */ -#include "ImageUtils.hpp" - -void convertImgIoInt8(void * data, const size_t sz) -{ - uint8_t * tmp_req_data = static_cast<uint8_t *>(data); - int8_t * tmp_signed_req_data = static_cast<int8_t *>(data); - - for (size_t i = 0; i < sz; ++i) { - tmp_signed_req_data[i] = static_cast<int8_t>( - static_cast<int32_t>(tmp_req_data[i]) - 128); - } -} - -void convertImgIoGreyscale(const uint8_t * srcPtr, uint8_t * dstPtr, const size_t sz) -{ - for (size_t i = 0; i < sz; ++i, srcPtr += 3) { - *dstPtr++ = 0.2989 * (*srcPtr) + - 0.587 * (*(srcPtr+1)) + - 0.114 * (*(srcPtr+2)); - } -}
\ No newline at end of file diff --git a/tests/utils/ImageUtils.hpp b/tests/utils/ImageUtils.hpp deleted file mode 100644 index 838dcef..0000000 --- a/tests/utils/ImageUtils.hpp +++ /dev/null @@ -1,26 +0,0 @@ -/* - * Copyright (c) 2021 Arm Limited. All rights reserved. - * SPDX-License-Identifier: Apache-2.0 - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - * See the License for the specific language governing permissions and - * limitations under the License. - */ -#ifndef IMAGEUTILS_HPP -#define IMAGEUTILS_HPP - -#include <catch.hpp> - -void convertImgIoInt8(void * data, const size_t sz); - -void convertImgIoGreyscale(const uint8_t * srcPtr, uint8_t * dstPtr, const size_t sz); - -#endif /* IMAGEUTILS_HPP */ |