mirror of
https://gitee.com/bianbu-linux/ai-support
synced 2025-04-24 22:27:13 -04:00
24 lines
698 B
C++
24 lines
698 B
C++
#ifndef SUPPORT_INCLUDE_TASK_VISION_IMAGE_CLASSIFICATION_TASK_H_
|
|
#define SUPPORT_INCLUDE_TASK_VISION_IMAGE_CLASSIFICATION_TASK_H_
|
|
|
|
#include <memory> // for: shared_ptr
|
|
#include <string>
|
|
|
|
#include "opencv2/opencv.hpp"
|
|
#include "task/vision/image_classification_types.h"
|
|
|
|
class imageClassificationTask {
|
|
public:
|
|
imageClassificationTask(const std::string& filePath,
|
|
const std::string& labelFilepath);
|
|
~imageClassificationTask() = default;
|
|
ImageClassificationResult Classify(const cv::Mat& img_raw);
|
|
int getInitFlag();
|
|
|
|
private:
|
|
class impl;
|
|
std::shared_ptr<impl> pimpl_;
|
|
int init_flag_;
|
|
};
|
|
|
|
#endif // SUPPORT_INCLUDE_TASK_VISION_IMAGE_CLASSIFICATION_TASK_H_
|