Skip to content

Commit d19098a

Browse files
committed
Fix anomaly task
Was missing reverse input channels
1 parent 9d44d95 commit d19098a

File tree

3 files changed

+13
-2
lines changed

3 files changed

+13
-2
lines changed

src/cpp/src/tasks/anomaly.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@ cv::Size Anomaly::serialize(std::shared_ptr<ov::Model>& ov_model) {
2323
std::vector<float> mean_values;
2424
if (ov_model->has_rt_info("model_info")) {
2525
auto config = ov_model->get_rt_info<ov::AnyMap>("model_info");
26+
reverse_input_channels = utils::get_from_any_maps("reverse_input_channels", config, ov::AnyMap{}, reverse_input_channels);
2627
scale_values = utils::get_from_any_maps("scale_values", config, ov::AnyMap{}, scale_values);
2728
mean_values = utils::get_from_any_maps("mean_values", config, ov::AnyMap{}, mean_values);
2829
}

src/cpp/src/utils/preprocessing.cpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,9 @@ std::shared_ptr<ov::Model> embedProcessing(std::shared_ptr<ov::Model>& model,
5454
ppp.input(inputName).preprocess().scale(scale);
5555
}
5656

57-
return ppp.build();
57+
auto ov_model = ppp.build();
58+
ov::set_batch(ov_model, 1);
59+
return ov_model;
5860
}
5961

6062
ov::preprocess::PostProcessSteps::CustomPostprocessOp createResizeGraph(RESIZE_MODE resizeMode,

tests/cpp/test_accuracy.cpp

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,15 @@ TEST_P(ModelParameterizedTest, AccuracyTest) {
104104
} else if (data.type == "MaskRCNNModel") {
105105
GTEST_SKIP();
106106
} else if (data.type == "AnomalyDetection") {
107-
GTEST_SKIP();
107+
auto model = Anomaly::load(model_path);
108+
109+
for (auto& test_data : data.test_data) {
110+
std::string image_path = DATA_DIR + '/' + test_data.image;
111+
cv::Mat image = cv::imread(image_path);
112+
auto result = model.infer(image);
113+
114+
EXPECT_EQ(std::string{result}, test_data.reference[0]);
115+
}
108116
} else {
109117
FAIL() << "No implementation for model type " << data.type;
110118
}

0 commit comments

Comments
 (0)