Skip to content

Commit 0579333

Browse files
committed
Fix anomaly task
Was missing reverse input channels
1 parent f26de53 commit 0579333

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
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
}

tests/cpp/test_accuracy.cpp

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,15 @@ TEST_P(ModelParameterizedTest, AccuracyTest) {
127127
EXPECT_EQ(std::string{result}, test_data.reference[0]);
128128
}
129129
} else if (data.type == "AnomalyDetection") {
130-
GTEST_SKIP();
130+
auto model = Anomaly::load(model_path);
131+
132+
for (auto& test_data : data.test_data) {
133+
std::string image_path = DATA_DIR + '/' + test_data.image;
134+
cv::Mat image = cv::imread(image_path);
135+
auto result = model.infer(image);
136+
137+
EXPECT_EQ(std::string{result}, test_data.reference[0]);
138+
}
131139
} else {
132140
FAIL() << "No implementation for model type " << data.type;
133141
}

0 commit comments

Comments
 (0)