Skip to content

Commit 73510f7

Browse files
committed
Prevent anymap from trying to parse u8 type
1 parent 23462bb commit 73510f7

File tree

5 files changed

+5
-5
lines changed

5 files changed

+5
-5
lines changed

src/cpp/src/tasks/anomaly.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ void Anomaly::serialize(std::shared_ptr<ov::Model>& ov_model) {
2929
if (ov_model->has_rt_info("model_info")) {
3030
auto config = ov_model->get_rt_info<ov::AnyMap>("model_info");
3131
resize_mode = utils::get_from_any_maps("resize_type", config, ov::AnyMap{}, resize_mode);
32-
pad_value = utils::get_from_any_maps("pad_value", config, ov::AnyMap{}, pad_value);
32+
pad_value = utils::get_from_any_maps<unsigned>("pad_value", config, ov::AnyMap{}, pad_value);
3333
reverse_input_channels =
3434
utils::get_from_any_maps("reverse_input_channels", config, ov::AnyMap{}, reverse_input_channels);
3535
scale_values = utils::get_from_any_maps("scale_values", config, ov::AnyMap{}, scale_values);

src/cpp/src/tasks/classification.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,7 @@ void Classification::serialize(std::shared_ptr<ov::Model>& ov_model) {
113113
scale_values = utils::get_from_any_maps("scale_values", config, ov::AnyMap{}, scale_values);
114114
mean_values = utils::get_from_any_maps("mean_values", config, ov::AnyMap{}, mean_values);
115115
uint8_t pad_value = 0;
116-
pad_value = utils::get_from_any_maps("pad_value", config, ov::AnyMap{}, pad_value);
116+
pad_value = utils::get_from_any_maps<unsigned>("pad_value", config, ov::AnyMap{}, pad_value);
117117
bool reverse_input_channels = false;
118118
reverse_input_channels =
119119
utils::get_from_any_maps("reverse_input_channels", config, ov::AnyMap{}, reverse_input_channels);

src/cpp/src/tasks/detection/ssd.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ void SSD::serialize(std::shared_ptr<ov::Model>& ov_model) {
9191

9292
auto input_shape = ov::Shape{shape[ov::layout::width_idx(layout)], shape[ov::layout::height_idx(layout)]};
9393
uint8_t pad_value = 0;
94-
pad_value = utils::get_from_any_maps("pad_value", config, ov::AnyMap{}, pad_value);
94+
pad_value = utils::get_from_any_maps<unsigned>("pad_value", config, ov::AnyMap{}, pad_value);
9595

9696
std::vector<float> scale_values;
9797
std::vector<float> mean_values;

src/cpp/src/tasks/instance_segmentation.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ void InstanceSegmentation::serialize(std::shared_ptr<ov::Model>& ov_model) {
149149
scale_values = utils::get_from_any_maps("scale_values", config, ov::AnyMap{}, scale_values);
150150
mean_values = utils::get_from_any_maps("mean_values", config, ov::AnyMap{}, mean_values);
151151
uint8_t pad_value = 0;
152-
pad_value = utils::get_from_any_maps("pad_value", config, ov::AnyMap{}, pad_value);
152+
pad_value = utils::get_from_any_maps<unsigned>("pad_value", config, ov::AnyMap{}, pad_value);
153153
bool reverse_input_channels = false;
154154
reverse_input_channels =
155155
utils::get_from_any_maps("reverse_input_channels", config, ov::AnyMap{}, reverse_input_channels);

src/cpp/src/tasks/semantic_segmentation.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@ void SemanticSegmentation::serialize(std::shared_ptr<ov::Model>& ov_model) {
8686
utils::RESIZE_MODE resize_mode = utils::RESIZE_MODE::RESIZE_FILL;
8787
resize_mode = utils::get_from_any_maps("resize_type", config, ov::AnyMap{}, resize_mode);
8888
uint8_t pad_value = 0;
89-
pad_value = utils::get_from_any_maps("pad_value", config, ov::AnyMap{}, pad_value);
89+
pad_value = utils::get_from_any_maps<unsigned>("pad_value", config, ov::AnyMap{}, pad_value);
9090
bool reverse_input_channels = false;
9191
reverse_input_channels =
9292
utils::get_from_any_maps("reverse_input_channels", config, ov::AnyMap{}, reverse_input_channels);

0 commit comments

Comments
 (0)