Skip to content

Commit ccbfcbb

Browse files
committed
Rename ImageModel to BaseModel
as opposed to ModelBase as it was. So first it was ImageModel: ModelBase. Then ModelBase merged into ImageModel now rename ImageModel to BaseModel
1 parent a068643 commit ccbfcbb

28 files changed

+115
-119
lines changed

src/cpp/models/include/models/anomaly_model.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,15 +4,15 @@
44
*/
55

66
#pragma once
7-
#include "models/image_model.h"
7+
#include "models/base_model.h"
88

99
namespace ov {
1010
class Model;
1111
} // namespace ov
1212
struct AnomalyResult;
1313
struct ImageInputData;
1414

15-
class AnomalyModel : public ImageModel {
15+
class AnomalyModel : public BaseModel {
1616
public:
1717
AnomalyModel(std::shared_ptr<ov::Model>& model, const ov::AnyMap& configuration);
1818
AnomalyModel(std::shared_ptr<InferenceAdapter>& adapter, const ov::AnyMap& configuration = {});

src/cpp/models/include/models/image_model.h renamed to src/cpp/models/include/models/base_model.h

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -23,19 +23,19 @@ struct InputData;
2323
struct InternalModelData;
2424

2525
// ImageModel implements preprocess(), ImageModel's direct or indirect children are expected to implement prostprocess()
26-
class ImageModel {
26+
class BaseModel {
2727
public:
2828
/// Constructor
2929
/// @param modelFile name of model to load
3030
/// @param useAutoResize - if true, image is resized by openvino
3131
/// @param layout - model input layout
32-
ImageModel(const std::string& modelFile,
33-
const std::string& resize_type,
34-
bool useAutoResize,
35-
const std::string& layout = "");
32+
BaseModel(const std::string& modelFile,
33+
const std::string& resize_type,
34+
bool useAutoResize,
35+
const std::string& layout = "");
3636

37-
ImageModel(std::shared_ptr<ov::Model>& model, const ov::AnyMap& configuration);
38-
ImageModel(std::shared_ptr<InferenceAdapter>& adapter, const ov::AnyMap& configuration = {});
37+
BaseModel(std::shared_ptr<ov::Model>& model, const ov::AnyMap& configuration);
38+
BaseModel(std::shared_ptr<InferenceAdapter>& adapter, const ov::AnyMap& configuration = {});
3939

4040
virtual std::shared_ptr<InternalModelData> preprocess(const InputData& inputData, InferenceInput& input);
4141
virtual std::unique_ptr<ResultBase> postprocess(InferenceResult& infResult) = 0;

src/cpp/models/include/models/classification_model.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
#include <unordered_map>
1313
#include <vector>
1414

15-
#include "models/image_model.h"
15+
#include "models/base_model.h"
1616

1717
namespace ov {
1818
class Model;
@@ -88,7 +88,7 @@ class ProbabilisticLabelsResolver : public GreedyLabelsResolver {
8888
SimpleLabelsGraph label_tree;
8989
};
9090

91-
class ClassificationModel : public ImageModel {
91+
class ClassificationModel : public BaseModel {
9292
public:
9393
ClassificationModel(std::shared_ptr<ov::Model>& model, const ov::AnyMap& configuration);
9494
ClassificationModel(std::shared_ptr<InferenceAdapter>& adapter, const ov::AnyMap& configuration = {});

src/cpp/models/include/models/detection_model.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,13 +7,13 @@
77

88
#include <string>
99

10-
#include "models/image_model.h"
10+
#include "models/base_model.h"
1111

1212
struct DetectionResult;
1313
struct ImageInputData;
1414
struct InferenceAdatper;
1515

16-
class DetectionModel : public ImageModel {
16+
class DetectionModel : public BaseModel {
1717
public:
1818
DetectionModel(std::shared_ptr<ov::Model>& model, const ov::AnyMap& configuration);
1919
DetectionModel(std::shared_ptr<InferenceAdapter>& adapter, const ov::AnyMap& configuration = {});

src/cpp/models/include/models/instance_segmentation.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
#include <string>
99
#include <vector>
1010

11-
#include "models/image_model.h"
11+
#include "models/base_model.h"
1212

1313
namespace ov {
1414
class Model;
@@ -19,7 +19,7 @@ struct InstanceSegmentationResult;
1919
struct ImageInputData;
2020
struct SegmentedObject;
2121

22-
class MaskRCNNModel : public ImageModel {
22+
class MaskRCNNModel : public BaseModel {
2323
public:
2424
MaskRCNNModel(std::shared_ptr<ov::Model>& model, const ov::AnyMap& configuration);
2525
MaskRCNNModel(std::shared_ptr<InferenceAdapter>& adapter, const ov::AnyMap& configuration = {});

src/cpp/models/include/models/keypoint_detection.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
#include <string>
99
#include <vector>
1010

11-
#include "models/image_model.h"
11+
#include "models/base_model.h"
1212

1313
namespace ov {
1414
class Model;
@@ -18,7 +18,7 @@ struct ResultBase;
1818
struct KeypointDetectionResult;
1919
struct ImageInputData;
2020

21-
class KeypointDetectionModel : public ImageModel {
21+
class KeypointDetectionModel : public BaseModel {
2222
public:
2323
KeypointDetectionModel(std::shared_ptr<ov::Model>& model, const ov::AnyMap& configuration);
2424
KeypointDetectionModel(std::shared_ptr<InferenceAdapter>& adapter, const ov::AnyMap& configuration = {});

src/cpp/models/include/models/segmentation_model.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
#include <string>
99
#include <vector>
1010

11-
#include "models/image_model.h"
11+
#include "models/base_model.h"
1212

1313
namespace ov {
1414
class Model;
@@ -20,7 +20,7 @@ struct ImageResultWithSoftPrediction;
2020
struct ImageInputData;
2121
struct Contour;
2222

23-
class SegmentationModel : public ImageModel {
23+
class SegmentationModel : public BaseModel {
2424
public:
2525
SegmentationModel(std::shared_ptr<ov::Model>& model, const ov::AnyMap& configuration);
2626
SegmentationModel(std::shared_ptr<InferenceAdapter>& adapter, const ov::AnyMap& configuration = {});

src/cpp/models/src/anomaly_model.cpp

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
#include <openvino/core/model.hpp>
1111
#include <ostream>
1212

13-
#include "models/image_model.h"
13+
#include "models/base_model.h"
1414
#include "models/input_data.h"
1515
#include "models/internal_model_data.h"
1616
#include "models/results.h"
@@ -29,23 +29,23 @@ void AnomalyModel::init_from_config(const ov::AnyMap& top_priority, const ov::An
2929
}
3030

3131
AnomalyModel::AnomalyModel(std::shared_ptr<ov::Model>& model, const ov::AnyMap& configuration)
32-
: ImageModel(model, configuration) {
32+
: BaseModel(model, configuration) {
3333
init_from_config(configuration, model->get_rt_info<ov::AnyMap>("model_info"));
3434
}
3535

3636
AnomalyModel::AnomalyModel(std::shared_ptr<InferenceAdapter>& adapter, const ov::AnyMap& configuration)
37-
: ImageModel(adapter, configuration) {
37+
: BaseModel(adapter, configuration) {
3838
init_from_config(configuration, adapter->getModelConfig());
3939
}
4040

4141
std::unique_ptr<AnomalyResult> AnomalyModel::infer(const ImageInputData& inputData) {
42-
auto result = ImageModel::inferImage(inputData);
42+
auto result = BaseModel::inferImage(inputData);
4343

4444
return std::unique_ptr<AnomalyResult>(static_cast<AnomalyResult*>(result.release()));
4545
}
4646

4747
std::vector<std::unique_ptr<AnomalyResult>> AnomalyModel::inferBatch(const std::vector<ImageInputData>& inputImgs) {
48-
auto results = ImageModel::inferBatchImage(inputImgs);
48+
auto results = BaseModel::inferBatchImage(inputImgs);
4949
std::vector<std::unique_ptr<AnomalyResult>> anoResults;
5050
anoResults.reserve(results.size());
5151
for (auto& result : results) {
@@ -177,7 +177,7 @@ void AnomalyModel::prepareInputsOutputs(std::shared_ptr<ov::Model>& model) {
177177
const ov::Layout& inputLayout = getInputLayout(input);
178178

179179
if (!embedded_processing) {
180-
model = ImageModel::embedProcessing(
180+
model = BaseModel::embedProcessing(
181181
model,
182182
inputNames[0],
183183
inputLayout,
@@ -194,7 +194,7 @@ void AnomalyModel::prepareInputsOutputs(std::shared_ptr<ov::Model>& model) {
194194
}
195195

196196
void AnomalyModel::updateModelInfo() {
197-
ImageModel::updateModelInfo();
197+
BaseModel::updateModelInfo();
198198

199199
model->set_rt_info(AnomalyModel::ModelType, "model_info", "model_type");
200200
model->set_rt_info(task, "model_info", "task");

0 commit comments

Comments
 (0)