Skip to content

Commit 4cf6e75

Browse files
committed
Expose inference mode to all tilers
1 parent 9b5d37c commit 4cf6e75

File tree

6 files changed

+15
-9
lines changed

6 files changed

+15
-9
lines changed

model_api/cpp/tilers/include/tilers/detection.h

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,8 @@ struct DetectionResult;
1010

1111
class DetectionTiler : public TilerBase {
1212
public:
13-
DetectionTiler(const std::shared_ptr<ImageModel>& model, const ov::AnyMap& configuration);
13+
DetectionTiler(const std::shared_ptr<ImageModel>& model, const ov::AnyMap& configuration,
14+
ExecutionMode exec_mode = ExecutionMode::sync);
1415
virtual ~DetectionTiler() = default;
1516

1617
virtual std::unique_ptr<DetectionResult> run(const ImageInputData& inputData);

model_api/cpp/tilers/include/tilers/instance_segmentation.h

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,8 @@ struct InstanceSegmentationResult;
1111
class InstanceSegmentationTiler : public TilerBase {
1212
/*InstanceSegmentationTiler tiler works with MaskRCNNModel model only*/
1313
public:
14-
InstanceSegmentationTiler(std::shared_ptr<ImageModel> model, const ov::AnyMap& configuration);
14+
InstanceSegmentationTiler(std::shared_ptr<ImageModel> model, const ov::AnyMap& configuration,
15+
ExecutionMode exec_mode = ExecutionMode::sync);
1516
virtual std::unique_ptr<InstanceSegmentationResult> run(const ImageInputData& inputData);
1617
virtual ~InstanceSegmentationTiler() = default;
1718
bool postprocess_semantic_masks = true;

model_api/cpp/tilers/include/tilers/semantic_segmentation.h

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,8 @@ struct ImageResultWithSoftPrediction;
1111

1212
class SemanticSegmentationTiler : public TilerBase {
1313
public:
14-
SemanticSegmentationTiler(std::shared_ptr<ImageModel> model, const ov::AnyMap& configuration);
14+
SemanticSegmentationTiler(std::shared_ptr<ImageModel> model, const ov::AnyMap& configuration,
15+
ExecutionMode exec_mode = ExecutionMode::sync);
1516
virtual std::unique_ptr<ImageResultWithSoftPrediction> run(const ImageInputData& inputData);
1617
virtual ~SemanticSegmentationTiler() = default;
1718

model_api/cpp/tilers/src/detection.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,8 +27,9 @@ cv::Mat non_linear_normalization(cv::Mat& class_map) {
2727

2828
} // namespace
2929

30-
DetectionTiler::DetectionTiler(const std::shared_ptr<ImageModel>& _model, const ov::AnyMap& configuration)
31-
: TilerBase(_model, configuration) {
30+
DetectionTiler::DetectionTiler(const std::shared_ptr<ImageModel>& _model, const ov::AnyMap& configuration,
31+
ExecutionMode exec_mode)
32+
: TilerBase(_model, configuration, exec_mode) {
3233
ov::AnyMap extra_config;
3334
try {
3435
auto ov_model = model->getModel();

model_api/cpp/tilers/src/instance_segmentation.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,9 @@ class MaskRCNNModelParamsSetter {
3333
} // namespace
3434

3535
InstanceSegmentationTiler::InstanceSegmentationTiler(std::shared_ptr<ImageModel> _model,
36-
const ov::AnyMap& configuration)
37-
: TilerBase(_model, configuration) {
36+
const ov::AnyMap& configuration,
37+
ExecutionMode exec_mode)
38+
: TilerBase(_model, configuration, exec_mode) {
3839
ov::AnyMap extra_config;
3940
try {
4041
auto ov_model = model->getModel();

model_api/cpp/tilers/src/semantic_segmentation.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,9 @@ void normalize_soft_prediction(cv::Mat& soft_prediction, const cv::Mat& normaliz
3333
} // namespace
3434

3535
SemanticSegmentationTiler::SemanticSegmentationTiler(std::shared_ptr<ImageModel> _model,
36-
const ov::AnyMap& configuration)
37-
: TilerBase(_model, configuration) {
36+
const ov::AnyMap& configuration,
37+
ExecutionMode exec_mode)
38+
: TilerBase(_model, configuration, exec_mode) {
3839
ov::AnyMap extra_config;
3940
try {
4041
auto ov_model = model->getModel();

0 commit comments

Comments
 (0)