mv_machine_learning: convert ImageClassification class into template class

[platform/core/api/mediavision.git] / mv_machine_learning / image_classification / include / image_classification_adapter.h

diff --git a/mv_machine_learning/image_classification/include/image_classification_adapter.h b/mv_machine_learning/image_classification/include/image_classification_adapter.h
index e625efe..d2f24d0 100644 (file)
--- a/mv_machine_learning/image_classification/include/image_classification_adapter.h
+++ b/mv_machine_learning/image_classification/include/image_classification_adapter.h
@@ -23,6 +23,7 @@
 #include "itask.h"
 #include "machine_learning_config.h"
 #include "image_classification_default.h"
+#include "iimage_classification.h"
 
 namespace mediavision
 {
@@ -31,7 +32,7 @@ namespace machine_learning
 template<typename T, typename V> class ImageClassificationAdapter : public mediavision::common::ITask<T, V>
 {
 private:
-       std::unique_ptr<ImageClassification> _image_classification;
+       std::unique_ptr<IImageClassification> _image_classification;
        std::shared_ptr<MachineLearningConfig> _config;
        T _source;
        const std::string _config_file_name = "image_classification.json";