[C API][Preprocess] add more preprocess api (#13770)
* [C API][Preprocess API] Add api reverse_channels(), convert_layout(), crop(), scale(), mean() Signed-off-by: xuejun <Xuejun.Zhai@intel.com> * [C API][Preprocess API] Add unite test case for api reverse_channels(), convert_layout(), crop(), scale(), mean() Signed-off-by: xuejun <Xuejun.Zhai@intel.com> * [C API][Preprocess API] code format api reverse_channels(), convert_layout(), crop(), scale(), mean() Signed-off-by: xuejun <Xuejun.Zhai@intel.com> * [C API][Preprocess API] code format api reverse_channels(), convert_layout(), crop(), scale(), mean() Signed-off-by: xuejun <Xuejun.Zhai@intel.com> * [C API][Preprocess API] code format Signed-off-by: xuejun <Xuejun.Zhai@intel.com> * [C API][PPP Test Case] clarify code with TEST_F Signed-off-by: xuejun <Xuejun.Zhai@intel.com> * [C API][PPP Test Case] clarify code Signed-off-by: xuejun <Xuejun.Zhai@intel.com> * [C API][PPP Test Case] remove space Signed-off-by: xuejun <Xuejun.Zhai@intel.com> * [C API][PPP Test Case] fix code format issue Signed-off-by: xuejun <Xuejun.Zhai@intel.com> * [C API][PPP Test Case] fix code format issue Signed-off-by: xuejun <Xuejun.Zhai@intel.com> * [C API][PPP Test Case] fix code format issue Signed-off-by: xuejun <Xuejun.Zhai@intel.com> Signed-off-by: xuejun <Xuejun.Zhai@intel.com>
This commit is contained in:
Xuejun Zhai authored and 
GitHub committed
1 parent
f488e6cc49
commit
2faddf61bf
3 files changed
+268
-403
No files matched your search
@@ -212,6 +212,62 @@ ov_preprocess_preprocess_steps_resize(
|
||||
ov_preprocess_preprocess_steps_t* preprocess_input_process_steps,
|
||||
const ov_preprocess_resize_algorithm_e resize_algorithm);
|
||||
|
||||
/**
|
||||
* @brief Add scale preprocess operation. Divide each element of input by specified value.
|
||||
* @ingroup prepostprocess
|
||||
* @param preprocess_input_process_steps A pointer to ov_preprocess_preprocess_steps_t.
|
||||
* @param value Scaling value
|
||||
* @return Status code of the operation: OK(0) for success.
|
||||
*/
|
||||
OPENVINO_C_API(ov_status_e)
|
||||
ov_preprocess_preprocess_steps_scale(ov_preprocess_preprocess_steps_t* preprocess_input_process_steps, float value);
|
||||
|
||||
/**
|
||||
* @brief Add mean preprocess operation. Subtract specified value from each element of input.
|
||||
* @ingroup prepostprocess
|
||||
* @param preprocess_input_process_steps A pointer to ov_preprocess_preprocess_steps_t.
|
||||
* @param value Value to subtract from each element.
|
||||
* @return Status code of the operation: OK(0) for success.
|
||||
*/
|
||||
OPENVINO_C_API(ov_status_e)
|
||||
ov_preprocess_preprocess_steps_mean(ov_preprocess_preprocess_steps_t* preprocess_input_process_steps, float value);
|
||||
|
||||
/**
|
||||
* @brief Crop input tensor between begin and end coordinates.
|
||||
* @ingroup prepostprocess
|
||||
* @param preprocess_input_process_steps A pointer to ov_preprocess_preprocess_steps_t.
|
||||
* @param begin Pointer to begin indexes for input tensor cropping.
|
||||
* Negative values represent counting elements from the end of input tensor
|
||||
* @param begin_size The size of begin array
|
||||
* @param end Pointer to end indexes for input tensor cropping.
|
||||
* End indexes are exclusive, which means values including end edge are not included in the output slice.
|
||||
* Negative values represent counting elements from the end of input tensor
|
||||
* @param end_size The size of end array
|
||||
* @return Status code of the operation: OK(0) for success.
|
||||
*/
|
||||
OPENVINO_C_API(ov_status_e)
|
||||
ov_preprocess_preprocess_steps_crop(ov_preprocess_preprocess_steps_t* preprocess_input_process_steps,
|
||||
int32_t* begin, int32_t begin_size, int32_t* end, int32_t end_size);
|
||||
|
||||
/**
|
||||
* @brief Add 'convert layout' operation to specified layout.
|
||||
* @ingroup prepostprocess
|
||||
* @param preprocess_input_process_steps A pointer to ov_preprocess_preprocess_steps_t.
|
||||
* @param layout A point to ov_layout_t
|
||||
* @return Status code of the operation: OK(0) for success.
|
||||
*/
|
||||
OPENVINO_C_API(ov_status_e)
|
||||
ov_preprocess_preprocess_steps_convert_layout(ov_preprocess_preprocess_steps_t* preprocess_input_process_steps, ov_layout_t* layout);
|
||||
|
||||
/**
|
||||
* @brief Reverse channels operation.
|
||||
* @ingroup prepostprocess
|
||||
* @param preprocess_input_process_steps A pointer to ov_preprocess_preprocess_steps_t.
|
||||
* @return Status code of the operation: OK(0) for success.
|
||||
*/
|
||||
OPENVINO_C_API(ov_status_e)
|
||||
ov_preprocess_preprocess_steps_reverse_channels(ov_preprocess_preprocess_steps_t* preprocess_input_process_steps);
|
||||
|
||||
/**
|
||||
* @brief Set ov_preprocess_input_tensor_info_t precesion.
|
||||
* @ingroup ov_prepostprocess_c_api
|
||||
|
||||
@@ -151,6 +151,77 @@ ov_status_e ov_preprocess_preprocess_steps_resize(ov_preprocess_preprocess_steps
|
||||
return ov_status_e::OK;
|
||||
}
|
||||
|
||||
ov_status_e ov_preprocess_preprocess_steps_scale(ov_preprocess_preprocess_steps_t* preprocess_input_process_steps,
|
||||
float value) {
|
||||
if (!preprocess_input_process_steps) {
|
||||
return ov_status_e::INVALID_C_PARAM;
|
||||
}
|
||||
try {
|
||||
preprocess_input_process_steps->object->scale(value);
|
||||
}
|
||||
CATCH_OV_EXCEPTIONS
|
||||
|
||||
return ov_status_e::OK;
|
||||
}
|
||||
|
||||
ov_status_e ov_preprocess_preprocess_steps_mean(ov_preprocess_preprocess_steps_t* preprocess_input_process_steps,
|
||||
float value) {
|
||||
if (!preprocess_input_process_steps) {
|
||||
return ov_status_e::INVALID_C_PARAM;
|
||||
}
|
||||
try {
|
||||
preprocess_input_process_steps->object->mean(value);
|
||||
}
|
||||
CATCH_OV_EXCEPTIONS
|
||||
|
||||
return ov_status_e::OK;
|
||||
}
|
||||
|
||||
ov_status_e ov_preprocess_preprocess_steps_crop(ov_preprocess_preprocess_steps_t* preprocess_input_process_steps,
|
||||
int32_t* begin,
|
||||
int32_t begin_size,
|
||||
int32_t* end,
|
||||
int32_t end_size) {
|
||||
if (!preprocess_input_process_steps) {
|
||||
return ov_status_e::INVALID_C_PARAM;
|
||||
}
|
||||
try {
|
||||
std::vector<int32_t> vec_begin(begin, begin + begin_size);
|
||||
std::vector<int32_t> vec_end(end, end + end_size);
|
||||
preprocess_input_process_steps->object->crop(vec_begin, vec_end);
|
||||
}
|
||||
CATCH_OV_EXCEPTIONS
|
||||
|
||||
return ov_status_e::OK;
|
||||
}
|
||||
|
||||
ov_status_e ov_preprocess_preprocess_steps_convert_layout(
|
||||
ov_preprocess_preprocess_steps_t* preprocess_input_process_steps,
|
||||
ov_layout_t* layout) {
|
||||
if (!preprocess_input_process_steps || !layout) {
|
||||
return ov_status_e::INVALID_C_PARAM;
|
||||
}
|
||||
try {
|
||||
preprocess_input_process_steps->object->convert_layout(layout->object);
|
||||
}
|
||||
CATCH_OV_EXCEPTIONS
|
||||
|
||||
return ov_status_e::OK;
|
||||
}
|
||||
|
||||
ov_status_e ov_preprocess_preprocess_steps_reverse_channels(
|
||||
ov_preprocess_preprocess_steps_t* preprocess_input_process_steps) {
|
||||
if (!preprocess_input_process_steps) {
|
||||
return ov_status_e::INVALID_C_PARAM;
|
||||
}
|
||||
try {
|
||||
preprocess_input_process_steps->object->reverse_channels();
|
||||
}
|
||||
CATCH_OV_EXCEPTIONS
|
||||
|
||||
return ov_status_e::OK;
|
||||
}
|
||||
|
||||
ov_status_e ov_preprocess_input_tensor_info_set_element_type(
|
||||
ov_preprocess_input_tensor_info_t* preprocess_input_tensor_info,
|
||||
const ov_element_type_e element_type) {
|
||||
|
||||
@@ -3,231 +3,211 @@
|
||||
//
|
||||
#include "ov_test.hpp"
|
||||
|
||||
TEST(ov_preprocess, ov_preprocess_prepostprocessor_create) {
|
||||
ov_core_t* core = nullptr;
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
class ov_preprocess : public ::testing::Test {
|
||||
protected:
|
||||
void SetUp() override {
|
||||
core = nullptr;
|
||||
model = nullptr;
|
||||
preprocess = nullptr;
|
||||
input_info = nullptr;
|
||||
input_tensor_info = nullptr;
|
||||
input_process = nullptr;
|
||||
tensor = nullptr;
|
||||
output_info = nullptr;
|
||||
output_tensor_info = nullptr;
|
||||
input_model = nullptr;
|
||||
|
||||
ov_model_t* model = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
|
||||
ov_preprocess_prepostprocessor_t* preprocess = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
}
|
||||
void TearDown() override {
|
||||
ov_preprocess_input_model_info_free(input_model);
|
||||
ov_preprocess_output_tensor_info_free(output_tensor_info);
|
||||
ov_preprocess_output_info_free(output_info);
|
||||
ov_tensor_free(tensor);
|
||||
ov_preprocess_preprocess_steps_free(input_process);
|
||||
ov_preprocess_input_tensor_info_free(input_tensor_info);
|
||||
ov_preprocess_input_info_free(input_info);
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
}
|
||||
|
||||
public:
|
||||
ov_core_t* core;
|
||||
ov_model_t* model;
|
||||
ov_preprocess_prepostprocessor_t* preprocess;
|
||||
ov_preprocess_input_info_t* input_info;
|
||||
ov_preprocess_input_tensor_info_t* input_tensor_info;
|
||||
ov_preprocess_preprocess_steps_t* input_process;
|
||||
ov_tensor_t* tensor;
|
||||
ov_preprocess_output_info_t* output_info;
|
||||
ov_preprocess_output_tensor_info_t* output_tensor_info;
|
||||
ov_preprocess_input_model_info_t* input_model;
|
||||
};
|
||||
|
||||
TEST_F(ov_preprocess, ov_preprocess_prepostprocessor_create) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
}
|
||||
|
||||
TEST(ov_preprocess, ov_preprocess_prepostprocessor_get_input_info) {
|
||||
ov_core_t* core = nullptr;
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
|
||||
ov_model_t* model = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
|
||||
ov_preprocess_prepostprocessor_t* preprocess = nullptr;
|
||||
TEST_F(ov_preprocess, ov_preprocess_prepostprocessor_get_input_info) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
ov_preprocess_input_info_t* input_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_input_info(preprocess, &input_info));
|
||||
EXPECT_NE(nullptr, input_info);
|
||||
|
||||
ov_preprocess_input_info_free(input_info);
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
}
|
||||
|
||||
TEST(ov_preprocess, ov_preprocess_prepostprocessor_get_input_info_by_name) {
|
||||
ov_core_t* core = nullptr;
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
|
||||
ov_model_t* model = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
|
||||
ov_preprocess_prepostprocessor_t* preprocess = nullptr;
|
||||
TEST_F(ov_preprocess, ov_preprocess_prepostprocessor_get_input_info_by_name) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
ov_preprocess_input_info_t* input_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_input_info_by_name(preprocess, "data", &input_info));
|
||||
EXPECT_NE(nullptr, input_info);
|
||||
|
||||
ov_preprocess_input_info_free(input_info);
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
}
|
||||
|
||||
TEST(ov_preprocess, ov_preprocess_prepostprocessor_get_input_info_by_index) {
|
||||
ov_core_t* core = nullptr;
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
|
||||
ov_model_t* model = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
|
||||
ov_preprocess_prepostprocessor_t* preprocess = nullptr;
|
||||
TEST_F(ov_preprocess, ov_preprocess_prepostprocessor_get_input_info_by_index) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
ov_preprocess_input_info_t* input_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_input_info_by_index(preprocess, 0, &input_info));
|
||||
EXPECT_NE(nullptr, input_info);
|
||||
|
||||
ov_preprocess_input_info_free(input_info);
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
}
|
||||
|
||||
TEST(ov_preprocess, ov_preprocess_input_info_get_tensor_info) {
|
||||
ov_core_t* core = nullptr;
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
|
||||
ov_model_t* model = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
|
||||
ov_preprocess_prepostprocessor_t* preprocess = nullptr;
|
||||
TEST_F(ov_preprocess, ov_preprocess_input_info_get_tensor_info) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
ov_preprocess_input_info_t* input_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_input_info_by_index(preprocess, 0, &input_info));
|
||||
EXPECT_NE(nullptr, input_info);
|
||||
|
||||
ov_preprocess_input_tensor_info_t* input_tensor_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_input_info_get_tensor_info(input_info, &input_tensor_info));
|
||||
EXPECT_NE(nullptr, input_tensor_info);
|
||||
|
||||
ov_preprocess_input_tensor_info_free(input_tensor_info);
|
||||
ov_preprocess_input_info_free(input_info);
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
}
|
||||
|
||||
TEST(ov_preprocess, ov_preprocess_input_info_get_preprocess_steps) {
|
||||
ov_core_t* core = nullptr;
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
|
||||
ov_model_t* model = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
|
||||
ov_preprocess_prepostprocessor_t* preprocess = nullptr;
|
||||
TEST_F(ov_preprocess, ov_preprocess_input_info_get_preprocess_steps) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
ov_preprocess_input_info_t* input_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_input_info_by_index(preprocess, 0, &input_info));
|
||||
EXPECT_NE(nullptr, input_info);
|
||||
|
||||
ov_preprocess_preprocess_steps_t* input_process = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_input_info_get_preprocess_steps(input_info, &input_process));
|
||||
EXPECT_NE(nullptr, input_process);
|
||||
|
||||
ov_preprocess_preprocess_steps_free(input_process);
|
||||
ov_preprocess_input_info_free(input_info);
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
}
|
||||
|
||||
TEST(ov_preprocess, ov_preprocess_preprocess_steps_resize) {
|
||||
ov_core_t* core = nullptr;
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
|
||||
ov_model_t* model = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
|
||||
ov_preprocess_prepostprocessor_t* preprocess = nullptr;
|
||||
TEST_F(ov_preprocess, ov_preprocess_preprocess_steps_resize) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
ov_preprocess_input_info_t* input_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_input_info_by_index(preprocess, 0, &input_info));
|
||||
EXPECT_NE(nullptr, input_info);
|
||||
|
||||
ov_preprocess_preprocess_steps_t* input_process = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_input_info_get_preprocess_steps(input_info, &input_process));
|
||||
EXPECT_NE(nullptr, input_process);
|
||||
|
||||
OV_EXPECT_OK(ov_preprocess_preprocess_steps_resize(input_process, ov_preprocess_resize_algorithm_e::RESIZE_LINEAR));
|
||||
|
||||
ov_preprocess_preprocess_steps_free(input_process);
|
||||
ov_preprocess_input_info_free(input_info);
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
}
|
||||
|
||||
TEST(ov_preprocess, ov_preprocess_input_tensor_info_set_element_type) {
|
||||
ov_core_t* core = nullptr;
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
|
||||
ov_model_t* model = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
|
||||
ov_preprocess_prepostprocessor_t* preprocess = nullptr;
|
||||
TEST_F(ov_preprocess, ov_preprocess_preprocess_steps_scale) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_input_info_by_index(preprocess, 0, &input_info));
|
||||
EXPECT_NE(nullptr, input_info);
|
||||
|
||||
OV_EXPECT_OK(ov_preprocess_input_info_get_preprocess_steps(input_info, &input_process));
|
||||
EXPECT_NE(nullptr, input_process);
|
||||
|
||||
OV_EXPECT_OK(ov_preprocess_preprocess_steps_scale(input_process, 2.0f));
|
||||
}
|
||||
|
||||
TEST_F(ov_preprocess, ov_preprocess_preprocess_steps_mean) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_input_info_by_index(preprocess, 0, &input_info));
|
||||
EXPECT_NE(nullptr, input_info);
|
||||
|
||||
OV_EXPECT_OK(ov_preprocess_input_info_get_preprocess_steps(input_info, &input_process));
|
||||
EXPECT_NE(nullptr, input_process);
|
||||
|
||||
OV_EXPECT_OK(ov_preprocess_preprocess_steps_mean(input_process, 2.0f));
|
||||
}
|
||||
|
||||
TEST_F(ov_preprocess, ov_preprocess_preprocess_steps_crop) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_input_info_by_index(preprocess, 0, &input_info));
|
||||
EXPECT_NE(nullptr, input_info);
|
||||
|
||||
OV_EXPECT_OK(ov_preprocess_input_info_get_preprocess_steps(input_info, &input_process));
|
||||
EXPECT_NE(nullptr, input_process);
|
||||
|
||||
int32_t begin[] = {0, 0, 5, 10};
|
||||
int32_t end[] = {1, 3, 15, 20};
|
||||
OV_EXPECT_OK(ov_preprocess_preprocess_steps_crop(input_process, begin, 4, end, 4));
|
||||
}
|
||||
|
||||
TEST_F(ov_preprocess, ov_preprocess_preprocess_steps_convert_layout) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_input_info_by_index(preprocess, 0, &input_info));
|
||||
EXPECT_NE(nullptr, input_info);
|
||||
|
||||
OV_EXPECT_OK(ov_preprocess_input_info_get_preprocess_steps(input_info, &input_process));
|
||||
EXPECT_NE(nullptr, input_process);
|
||||
|
||||
ov_layout_t* layout = nullptr;
|
||||
const char* input_layout_desc = "NCHW";
|
||||
OV_EXPECT_OK(ov_layout_create(input_layout_desc, &layout));
|
||||
OV_EXPECT_OK(ov_preprocess_preprocess_steps_convert_layout(input_process, layout));
|
||||
|
||||
ov_layout_free(layout);
|
||||
}
|
||||
|
||||
TEST_F(ov_preprocess, ov_preprocess_preprocess_steps_reverse_channels) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_input_info_by_index(preprocess, 0, &input_info));
|
||||
EXPECT_NE(nullptr, input_info);
|
||||
|
||||
OV_EXPECT_OK(ov_preprocess_input_info_get_preprocess_steps(input_info, &input_process));
|
||||
EXPECT_NE(nullptr, input_process);
|
||||
|
||||
OV_EXPECT_OK(ov_preprocess_preprocess_steps_reverse_channels(input_process));
|
||||
}
|
||||
|
||||
TEST_F(ov_preprocess, ov_preprocess_input_tensor_info_set_element_type) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
ov_preprocess_input_info_t* input_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_input_info_by_index(preprocess, 0, &input_info));
|
||||
EXPECT_NE(nullptr, input_info);
|
||||
|
||||
ov_preprocess_input_tensor_info_t* input_tensor_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_input_info_get_tensor_info(input_info, &input_tensor_info));
|
||||
EXPECT_NE(nullptr, input_tensor_info);
|
||||
|
||||
OV_EXPECT_OK(ov_preprocess_input_tensor_info_set_element_type(input_tensor_info, ov_element_type_e::F32));
|
||||
|
||||
ov_preprocess_input_tensor_info_free(input_tensor_info);
|
||||
ov_preprocess_input_info_free(input_info);
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
}
|
||||
|
||||
TEST(ov_preprocess, ov_preprocess_input_tensor_info_set_from) {
|
||||
ov_core_t* core = nullptr;
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
|
||||
ov_model_t* model = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
|
||||
ov_preprocess_prepostprocessor_t* preprocess = nullptr;
|
||||
TEST_F(ov_preprocess, ov_preprocess_input_tensor_info_set_from) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
ov_preprocess_input_info_t* input_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_input_info_by_index(preprocess, 0, &input_info));
|
||||
EXPECT_NE(nullptr, input_info);
|
||||
|
||||
ov_preprocess_input_tensor_info_t* input_tensor_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_input_info_get_tensor_info(input_info, &input_tensor_info));
|
||||
EXPECT_NE(nullptr, input_tensor_info);
|
||||
|
||||
ov_tensor_t* tensor = nullptr;
|
||||
ov_shape_t shape;
|
||||
int64_t dims[4] = {1, 416, 416, 4};
|
||||
OV_EXPECT_OK(ov_shape_create(4, dims, &shape));
|
||||
@@ -235,32 +215,16 @@ TEST(ov_preprocess, ov_preprocess_input_tensor_info_set_from) {
|
||||
OV_EXPECT_OK(ov_tensor_create(ov_element_type_e::F32, shape, &tensor));
|
||||
OV_EXPECT_OK(ov_preprocess_input_tensor_info_set_from(input_tensor_info, tensor));
|
||||
|
||||
ov_preprocess_input_tensor_info_free(input_tensor_info);
|
||||
ov_preprocess_input_info_free(input_info);
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_shape_free(&shape);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
}
|
||||
|
||||
TEST(ov_preprocess, ov_preprocess_input_tensor_info_set_layout) {
|
||||
ov_core_t* core = nullptr;
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
|
||||
ov_model_t* model = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
|
||||
ov_preprocess_prepostprocessor_t* preprocess = nullptr;
|
||||
TEST_F(ov_preprocess, ov_preprocess_input_tensor_info_set_layout) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
ov_preprocess_input_info_t* input_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_input_info_by_index(preprocess, 0, &input_info));
|
||||
EXPECT_NE(nullptr, input_info);
|
||||
|
||||
ov_preprocess_input_tensor_info_t* input_tensor_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_input_info_get_tensor_info(input_info, &input_tensor_info));
|
||||
EXPECT_NE(nullptr, input_tensor_info);
|
||||
|
||||
@@ -269,63 +233,30 @@ TEST(ov_preprocess, ov_preprocess_input_tensor_info_set_layout) {
|
||||
OV_EXPECT_OK(ov_layout_create(input_layout_desc, &layout));
|
||||
OV_EXPECT_OK(ov_preprocess_input_tensor_info_set_layout(input_tensor_info, layout));
|
||||
|
||||
ov_preprocess_input_tensor_info_free(input_tensor_info);
|
||||
ov_preprocess_input_info_free(input_info);
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_layout_free(layout);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
}
|
||||
|
||||
TEST(ov_preprocess, ov_preprocess_input_tensor_info_set_color_format) {
|
||||
ov_core_t* core = nullptr;
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
|
||||
ov_model_t* model = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
|
||||
ov_preprocess_prepostprocessor_t* preprocess = nullptr;
|
||||
TEST_F(ov_preprocess, ov_preprocess_input_tensor_info_set_color_format) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
ov_preprocess_input_info_t* input_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_input_info_by_index(preprocess, 0, &input_info));
|
||||
EXPECT_NE(nullptr, input_info);
|
||||
|
||||
ov_preprocess_input_tensor_info_t* input_tensor_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_input_info_get_tensor_info(input_info, &input_tensor_info));
|
||||
EXPECT_NE(nullptr, input_tensor_info);
|
||||
|
||||
OV_EXPECT_OK(
|
||||
ov_preprocess_input_tensor_info_set_color_format(input_tensor_info, ov_color_format_e::NV12_SINGLE_PLANE));
|
||||
|
||||
ov_preprocess_input_tensor_info_free(input_tensor_info);
|
||||
ov_preprocess_input_info_free(input_info);
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
}
|
||||
|
||||
TEST(ov_preprocess, ov_preprocess_input_tensor_info_set_spatial_static_shape) {
|
||||
ov_core_t* core = nullptr;
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
|
||||
ov_model_t* model = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
|
||||
ov_preprocess_prepostprocessor_t* preprocess = nullptr;
|
||||
TEST_F(ov_preprocess, ov_preprocess_input_tensor_info_set_spatial_static_shape) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
ov_preprocess_input_info_t* input_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_input_info_by_index(preprocess, 0, &input_info));
|
||||
EXPECT_NE(nullptr, input_info);
|
||||
|
||||
ov_preprocess_input_tensor_info_t* input_tensor_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_input_info_get_tensor_info(input_info, &input_tensor_info));
|
||||
EXPECT_NE(nullptr, input_tensor_info);
|
||||
|
||||
@@ -333,260 +264,109 @@ TEST(ov_preprocess, ov_preprocess_input_tensor_info_set_spatial_static_shape) {
|
||||
size_t input_width = 500;
|
||||
OV_EXPECT_OK(
|
||||
ov_preprocess_input_tensor_info_set_spatial_static_shape(input_tensor_info, input_height, input_width));
|
||||
|
||||
ov_preprocess_input_tensor_info_free(input_tensor_info);
|
||||
ov_preprocess_input_info_free(input_info);
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
}
|
||||
|
||||
TEST(ov_preprocess, ov_preprocess_preprocess_steps_convert_element_type) {
|
||||
ov_core_t* core = nullptr;
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
|
||||
ov_model_t* model = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
|
||||
ov_preprocess_prepostprocessor_t* preprocess = nullptr;
|
||||
TEST_F(ov_preprocess, ov_preprocess_preprocess_steps_convert_element_type) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
ov_preprocess_input_info_t* input_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_input_info_by_index(preprocess, 0, &input_info));
|
||||
EXPECT_NE(nullptr, input_info);
|
||||
|
||||
ov_preprocess_preprocess_steps_t* input_process = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_input_info_get_preprocess_steps(input_info, &input_process));
|
||||
EXPECT_NE(nullptr, input_process);
|
||||
|
||||
ov_preprocess_input_tensor_info_t* input_tensor_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_input_info_get_tensor_info(input_info, &input_tensor_info));
|
||||
EXPECT_NE(nullptr, input_tensor_info);
|
||||
|
||||
OV_EXPECT_OK(ov_preprocess_input_tensor_info_set_element_type(input_tensor_info, ov_element_type_e::U8));
|
||||
OV_EXPECT_OK(ov_preprocess_preprocess_steps_convert_element_type(input_process, ov_element_type_e::F32));
|
||||
|
||||
ov_preprocess_input_tensor_info_free(input_tensor_info);
|
||||
ov_preprocess_preprocess_steps_free(input_process);
|
||||
ov_preprocess_input_info_free(input_info);
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
}
|
||||
|
||||
TEST(ov_preprocess, ov_preprocess_preprocess_steps_convert_color) {
|
||||
ov_core_t* core = nullptr;
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
|
||||
ov_model_t* model = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
|
||||
ov_preprocess_prepostprocessor_t* preprocess = nullptr;
|
||||
TEST_F(ov_preprocess, ov_preprocess_preprocess_steps_convert_color) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
ov_preprocess_input_info_t* input_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_input_info_by_index(preprocess, 0, &input_info));
|
||||
EXPECT_NE(nullptr, input_info);
|
||||
|
||||
ov_preprocess_preprocess_steps_t* input_process = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_input_info_get_preprocess_steps(input_info, &input_process));
|
||||
EXPECT_NE(nullptr, input_process);
|
||||
|
||||
ov_preprocess_input_tensor_info_t* input_tensor_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_input_info_get_tensor_info(input_info, &input_tensor_info));
|
||||
EXPECT_NE(nullptr, input_tensor_info);
|
||||
|
||||
OV_EXPECT_OK(
|
||||
ov_preprocess_input_tensor_info_set_color_format(input_tensor_info, ov_color_format_e::NV12_SINGLE_PLANE));
|
||||
OV_EXPECT_OK(ov_preprocess_preprocess_steps_convert_color(input_process, ov_color_format_e::BGR));
|
||||
|
||||
ov_preprocess_input_tensor_info_free(input_tensor_info);
|
||||
ov_preprocess_preprocess_steps_free(input_process);
|
||||
ov_preprocess_input_info_free(input_info);
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
}
|
||||
|
||||
TEST(ov_preprocess, ov_preprocess_prepostprocessor_get_output_info) {
|
||||
ov_core_t* core = nullptr;
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
|
||||
ov_model_t* model = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
|
||||
ov_preprocess_prepostprocessor_t* preprocess = nullptr;
|
||||
TEST_F(ov_preprocess, ov_preprocess_prepostprocessor_get_output_info) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
ov_preprocess_output_info_t* output_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_output_info(preprocess, &output_info));
|
||||
EXPECT_NE(nullptr, output_info);
|
||||
|
||||
ov_preprocess_output_info_free(output_info);
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
}
|
||||
|
||||
TEST(ov_preprocess, ov_preprocess_prepostprocessor_get_output_info_by_index) {
|
||||
ov_core_t* core = nullptr;
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
|
||||
ov_model_t* model = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
|
||||
ov_preprocess_prepostprocessor_t* preprocess = nullptr;
|
||||
TEST_F(ov_preprocess, ov_preprocess_prepostprocessor_get_output_info_by_index) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
ov_preprocess_output_info_t* output_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_output_info_by_index(preprocess, 0, &output_info));
|
||||
EXPECT_NE(nullptr, output_info);
|
||||
|
||||
ov_preprocess_output_info_free(output_info);
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
}
|
||||
|
||||
TEST(ov_preprocess, ov_preprocess_prepostprocessor_get_output_info_by_name) {
|
||||
ov_core_t* core = nullptr;
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
|
||||
ov_model_t* model = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
|
||||
ov_preprocess_prepostprocessor_t* preprocess = nullptr;
|
||||
TEST_F(ov_preprocess, ov_preprocess_prepostprocessor_get_output_info_by_name) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
ov_preprocess_output_info_t* output_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_output_info_by_name(preprocess, "fc_out", &output_info));
|
||||
EXPECT_NE(nullptr, output_info);
|
||||
|
||||
ov_preprocess_output_info_free(output_info);
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
}
|
||||
|
||||
TEST(ov_preprocess, ov_preprocess_output_info_get_tensor_info) {
|
||||
ov_core_t* core = nullptr;
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
|
||||
ov_model_t* model = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
|
||||
ov_preprocess_prepostprocessor_t* preprocess = nullptr;
|
||||
TEST_F(ov_preprocess, ov_preprocess_output_info_get_tensor_info) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
ov_preprocess_output_info_t* output_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_output_info_by_index(preprocess, 0, &output_info));
|
||||
EXPECT_NE(nullptr, output_info);
|
||||
|
||||
ov_preprocess_output_tensor_info_t* output_tensor_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_output_info_get_tensor_info(output_info, &output_tensor_info));
|
||||
EXPECT_NE(nullptr, output_tensor_info);
|
||||
|
||||
ov_preprocess_output_tensor_info_free(output_tensor_info);
|
||||
ov_preprocess_output_info_free(output_info);
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
}
|
||||
|
||||
TEST(ov_preprocess, ov_preprocess_output_set_element_type) {
|
||||
ov_core_t* core = nullptr;
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
|
||||
ov_model_t* model = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
|
||||
ov_preprocess_prepostprocessor_t* preprocess = nullptr;
|
||||
TEST_F(ov_preprocess, ov_preprocess_output_set_element_type) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
ov_preprocess_output_info_t* output_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_output_info_by_index(preprocess, 0, &output_info));
|
||||
EXPECT_NE(nullptr, output_info);
|
||||
|
||||
ov_preprocess_output_tensor_info_t* output_tensor_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_output_info_get_tensor_info(output_info, &output_tensor_info));
|
||||
EXPECT_NE(nullptr, output_tensor_info);
|
||||
|
||||
OV_EXPECT_OK(ov_preprocess_output_set_element_type(output_tensor_info, ov_element_type_e::F32));
|
||||
|
||||
ov_preprocess_output_tensor_info_free(output_tensor_info);
|
||||
ov_preprocess_output_info_free(output_info);
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
}
|
||||
|
||||
TEST(ov_preprocess, ov_preprocess_input_info_get_model_info) {
|
||||
ov_core_t* core = nullptr;
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
|
||||
ov_model_t* model = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
|
||||
ov_preprocess_prepostprocessor_t* preprocess = nullptr;
|
||||
TEST_F(ov_preprocess, ov_preprocess_input_info_get_model_info) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
ov_preprocess_input_info_t* input_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_input_info_by_index(preprocess, 0, &input_info));
|
||||
EXPECT_NE(nullptr, input_info);
|
||||
|
||||
ov_preprocess_input_model_info_t* input_model = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_input_info_get_model_info(input_info, &input_model));
|
||||
EXPECT_NE(nullptr, input_model);
|
||||
|
||||
ov_preprocess_input_model_info_free(input_model);
|
||||
ov_preprocess_input_info_free(input_info);
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
}
|
||||
|
||||
TEST(ov_preprocess, ov_preprocess_input_model_info_set_layout) {
|
||||
ov_core_t* core = nullptr;
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
|
||||
ov_model_t* model = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
|
||||
ov_preprocess_prepostprocessor_t* preprocess = nullptr;
|
||||
TEST_F(ov_preprocess, ov_preprocess_input_model_info_set_layout) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
ov_preprocess_input_info_t* input_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_input_info_by_index(preprocess, 0, &input_info));
|
||||
EXPECT_NE(nullptr, input_info);
|
||||
|
||||
ov_preprocess_input_model_info_t* input_model = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_input_info_get_model_info(input_info, &input_model));
|
||||
EXPECT_NE(nullptr, input_model);
|
||||
|
||||
@@ -594,25 +374,11 @@ TEST(ov_preprocess, ov_preprocess_input_model_info_set_layout) {
|
||||
const char* layout_desc = "NCHW";
|
||||
OV_EXPECT_OK(ov_layout_create(layout_desc, &layout));
|
||||
OV_EXPECT_OK(ov_preprocess_input_model_info_set_layout(input_model, layout));
|
||||
ov_layout_free(layout);
|
||||
|
||||
ov_preprocess_input_model_info_free(input_model);
|
||||
ov_preprocess_input_info_free(input_info);
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
ov_layout_free(layout);
|
||||
}
|
||||
|
||||
TEST(ov_preprocess, ov_preprocess_prepostprocessor_build) {
|
||||
ov_core_t* core = nullptr;
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
|
||||
ov_model_t* model = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
|
||||
ov_preprocess_prepostprocessor_t* preprocess = nullptr;
|
||||
TEST_F(ov_preprocess, ov_preprocess_prepostprocessor_build) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
@@ -621,32 +387,18 @@ TEST(ov_preprocess, ov_preprocess_prepostprocessor_build) {
|
||||
EXPECT_NE(nullptr, new_model);
|
||||
|
||||
ov_model_free(new_model);
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
}
|
||||
|
||||
TEST(ov_preprocess, ov_preprocess_prepostprocessor_build_apply) {
|
||||
ov_core_t* core = nullptr;
|
||||
OV_EXPECT_OK(ov_core_create(&core));
|
||||
EXPECT_NE(nullptr, core);
|
||||
|
||||
ov_model_t* model = nullptr;
|
||||
OV_EXPECT_OK(ov_core_read_model(core, xml, bin, &model));
|
||||
EXPECT_NE(nullptr, model);
|
||||
|
||||
ov_preprocess_prepostprocessor_t* preprocess = nullptr;
|
||||
TEST_F(ov_preprocess, ov_preprocess_prepostprocessor_build_apply) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_create(model, &preprocess));
|
||||
EXPECT_NE(nullptr, preprocess);
|
||||
|
||||
ov_preprocess_input_info_t* input_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_input_info_by_index(preprocess, 0, &input_info));
|
||||
EXPECT_NE(nullptr, input_info);
|
||||
|
||||
ov_preprocess_input_tensor_info_t* input_tensor_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_input_info_get_tensor_info(input_info, &input_tensor_info));
|
||||
EXPECT_NE(nullptr, input_tensor_info);
|
||||
ov_tensor_t* tensor = nullptr;
|
||||
|
||||
ov_shape_t shape;
|
||||
int64_t dims[4] = {1, 416, 416, 3};
|
||||
OV_EXPECT_OK(ov_shape_create(4, dims, &shape));
|
||||
@@ -660,12 +412,10 @@ TEST(ov_preprocess, ov_preprocess_prepostprocessor_build_apply) {
|
||||
OV_EXPECT_OK(ov_preprocess_input_tensor_info_set_layout(input_tensor_info, layout));
|
||||
ov_layout_free(layout);
|
||||
|
||||
ov_preprocess_preprocess_steps_t* input_process = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_input_info_get_preprocess_steps(input_info, &input_process));
|
||||
EXPECT_NE(nullptr, input_process);
|
||||
OV_EXPECT_OK(ov_preprocess_preprocess_steps_resize(input_process, ov_preprocess_resize_algorithm_e::RESIZE_LINEAR));
|
||||
|
||||
ov_preprocess_input_model_info_t* input_model = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_input_info_get_model_info(input_info, &input_model));
|
||||
EXPECT_NE(nullptr, input_model);
|
||||
|
||||
@@ -675,10 +425,8 @@ TEST(ov_preprocess, ov_preprocess_prepostprocessor_build_apply) {
|
||||
OV_EXPECT_OK(ov_preprocess_input_model_info_set_layout(input_model, model_layout));
|
||||
ov_layout_free(model_layout);
|
||||
|
||||
ov_preprocess_output_info_t* output_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_get_output_info_by_index(preprocess, 0, &output_info));
|
||||
EXPECT_NE(nullptr, output_info);
|
||||
ov_preprocess_output_tensor_info_t* output_tensor_info = nullptr;
|
||||
OV_EXPECT_OK(ov_preprocess_output_info_get_tensor_info(output_info, &output_tensor_info));
|
||||
EXPECT_NE(nullptr, output_tensor_info);
|
||||
OV_EXPECT_OK(ov_preprocess_output_set_element_type(output_tensor_info, ov_element_type_e::F32));
|
||||
@@ -687,16 +435,6 @@ TEST(ov_preprocess, ov_preprocess_prepostprocessor_build_apply) {
|
||||
OV_EXPECT_OK(ov_preprocess_prepostprocessor_build(preprocess, &new_model));
|
||||
EXPECT_NE(nullptr, new_model);
|
||||
|
||||
ov_preprocess_input_tensor_info_free(input_tensor_info);
|
||||
ov_tensor_free(tensor);
|
||||
ov_shape_free(&shape);
|
||||
ov_preprocess_preprocess_steps_free(input_process);
|
||||
ov_preprocess_input_model_info_free(input_model);
|
||||
ov_preprocess_output_tensor_info_free(output_tensor_info);
|
||||
ov_preprocess_output_info_free(output_info);
|
||||
ov_preprocess_input_info_free(input_info);
|
||||
ov_model_free(new_model);
|
||||
ov_preprocess_prepostprocessor_free(preprocess);
|
||||
ov_model_free(model);
|
||||
ov_core_free(core);
|
||||
}
|
||||
Reference in new issue
Block a user