整理目录结构,分离Task

This commit is contained in:
MistEO
2021-08-20 16:24:04 +08:00
parent 9d15b33b50
commit 2af89dfed6
46 changed files with 1717 additions and 1854 deletions

View File

@@ -0,0 +1,37 @@
#include "OcrAbstractTask.h"
#include <opencv2/opencv.hpp>
#include "AsstAux.h"
#include "Identify.h"
using namespace asst;
OcrAbstractTask::OcrAbstractTask(AsstCallback callback, void* callback_arg)
: AbstractTask(callback, callback_arg)
{
;
}
std::vector<TextArea> OcrAbstractTask::ocr_detect()
{
const cv::Mat& image = get_format_image();
return ocr_detect(image);
}
std::vector<TextArea> OcrAbstractTask::ocr_detect(const cv::Mat& image)
{
std::vector<TextArea> dst = m_identify_ptr->ocr_detect(image);
std::vector<json::value> all_text_json_vector;
for (const TextArea& text_area : dst) {
all_text_json_vector.emplace_back(Utf8ToGbk(text_area.text));
}
json::value all_text_json;
all_text_json["text"] = json::array(all_text_json_vector);
all_text_json["task_chain"] = m_task_chain;
m_callback(AsstMsg::TextDetected, all_text_json, m_callback_arg);
return dst;
}