Skip to content

Commit 4f3242a

Browse files
authored
Expose inference mode to all tilers (#234)
* Expose inference mode to all tilers * Fix formatting
1 parent 9b5d37c commit 4f3242a

File tree

6 files changed

+19
-9
lines changed

6 files changed

+19
-9
lines changed

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

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,9 @@ 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,
14+
const ov::AnyMap& configuration,
15+
ExecutionMode exec_mode = ExecutionMode::sync);
1416
virtual ~DetectionTiler() = default;
1517

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

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

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,9 @@ 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,
15+
const ov::AnyMap& configuration,
16+
ExecutionMode exec_mode = ExecutionMode::sync);
1517
virtual std::unique_ptr<InstanceSegmentationResult> run(const ImageInputData& inputData);
1618
virtual ~InstanceSegmentationTiler() = default;
1719
bool postprocess_semantic_masks = true;

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

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,9 @@ 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,
15+
const ov::AnyMap& configuration,
16+
ExecutionMode exec_mode = ExecutionMode::sync);
1517
virtual std::unique_ptr<ImageResultWithSoftPrediction> run(const ImageInputData& inputData);
1618
virtual ~SemanticSegmentationTiler() = default;
1719

model_api/cpp/tilers/src/detection.cpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,8 +27,10 @@ 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,
31+
const ov::AnyMap& configuration,
32+
ExecutionMode exec_mode)
33+
: TilerBase(_model, configuration, exec_mode) {
3234
ov::AnyMap extra_config;
3335
try {
3436
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)