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
3131AnomalyModel::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
3636AnomalyModel::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
4141std::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
4747std::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
196196void 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