Skip to content

Commit f61a18d

Browse files
committed
Read resize from config in ssd
1 parent c6cb323 commit f61a18d

File tree

1 file changed

+1
-0
lines changed
  • src/cpp/include/tasks/detection

1 file changed

+1
-0
lines changed

src/cpp/include/tasks/detection/ssd.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@ class SSD {
3434
confidence_threshold = utils::get_from_any_maps("confidence_threshold", config, {}, confidence_threshold);
3535
input_shape.width = utils::get_from_any_maps("orig_width", config, {}, input_shape.width);
3636
input_shape.height = utils::get_from_any_maps("orig_height", config, {}, input_shape.height);
37+
resize_mode = utils::get_from_any_maps("resize_type", user_config, model_config, resize_mode);
3738
}
3839
std::map<std::string, ov::Tensor> preprocess(cv::Mat);
3940
DetectionResult postprocess(InferenceResult& infResult);

0 commit comments

Comments
 (0)