change recognizer output format

master
UnknownObject 2 months ago
parent d070976d35
commit 521273638d

@ -37,8 +37,8 @@ int main()
auto results = uns::EasyOCR_CPP::FullAuto(img);
for (const auto& [index, info] : results)
{
const auto& [text, conf] = info;
std::cout << "Box " << index << ": \"" << uns::EOCR_SupportTools::WtoA(text) << "\" Confidence=" << conf << "\n";
const auto& [text, conf, rect] = info;
printf("Result %d: Text = {%s}, Box = [%d, %d, %d, %d], Confidence = %.3f\n", (int)index, uns::EOCR_SupportTools::WtoA(text).c_str(), rect.tl().x, rect.tl().y, rect.br().x, rect.br().y, conf);
}
uns::EasyOCR_CPP::CleanupOCR();

@ -32,8 +32,9 @@ namespace uns
EN_CH
};
using EOCR_Result = std::pair<std::wstring, float>;
using EOCR_ResultSet = std::map<size_t, EOCR_Result>;
using EOCR_Result = std::tuple<std::wstring, float>;
using EOCR_Result_Single = std::tuple<std::wstring, float, cv::Rect>;
using EOCR_ResultSet = std::map<size_t, EOCR_Result_Single>;
using EOCRD_Rects = std::vector<std::vector<cv::Point2f>>;
}

Binary file not shown.

Binary file not shown.

@ -45,6 +45,8 @@ int main()
cv::blur(image, image, { 3,3 });
//WarmUP GPU
if (uns::G_OCRConfig.GetGPUUsage() != uns::OCRConfig::GPUUsage::CPUOnly)
{
for (int i = 0; i < 10; i++)
{
std::cout << "Warming UP GPU (" << i << ") ......" << std::endl;
@ -55,6 +57,7 @@ int main()
std::cout << "Warm UP [" << i << "] Cost " << static_cast<double>(end - start) / static_cast<double>(CLOCKS_PER_SEC) << " Second(s)" << std::endl;
}
std::cout << "GPU Warm UP Finished" << std::endl;
}
//Begin OCR
clock_t start = clock();
@ -64,8 +67,8 @@ int main()
clock_t recognize_finished = clock();
for (const auto& [index, info] : results)
{
const auto& [text, conf] = info;
std::cout << "Box " << index << ": \"" << WtoA(text) << "\" Confidence=" << conf << "\n";
const auto& [text, conf, rect] = info;
printf("Result %d: Text = {%s}, Box = [%d, %d, %d, %d], Confidence = %.3f\n", (int)index, WtoA(text).c_str(), rect.tl().x, rect.tl().y, rect.br().x, rect.br().y, conf);
}
//Output Time Cost

@ -377,7 +377,7 @@ uns::EOCR_ResultSet uns::EasyOCR_Recognizer::operator()(const cv::Mat& full_imag
if (crop.empty())
continue;
auto [text, conf] = (*this)(crop);
result_set.insert({ i, { text, conf } });
result_set.insert({ i, { text, conf, rect } });
}
return result_set;
}

@ -11,8 +11,9 @@ namespace uns
using VecFloat = std::vector<float>;
using IONames = std::vector<const char*>;
using IONamesStorage = std::vector<std::string>;
using EOCR_Result = std::pair<std::wstring, float>;
using EOCR_ResultSet = std::map<size_t, EOCR_Result>;
using EOCR_Result = std::tuple<std::wstring, float>;
using EOCR_Result_Single = std::tuple<std::wstring, float, cv::Rect>;
using EOCR_ResultSet = std::map<size_t, EOCR_Result_Single>;
using EOCRD_Rects = std::vector<std::vector<cv::Point2f>>;
class OCRToolBox

@ -378,7 +378,7 @@ uns::EOCR_ResultSet uns::EasyOCR_Recognizer::operator()(const cv::Mat& full_imag
if (crop.empty())
continue;
auto [text, conf] = (*this)(crop);
result_set.insert({ i, { text, conf } });
result_set.insert({ i, { text, conf, rect } });
}
return result_set;
}

@ -10,8 +10,9 @@ namespace uns
{
namespace easyocr
{
using EOCR_Result = std::pair<std::wstring, float>;
using EOCR_ResultSet = std::map<size_t, EOCR_Result>;
using EOCR_Result = std::tuple<std::wstring, float>;
using EOCR_Result_Single = std::tuple<std::wstring, float, cv::Rect>;
using EOCR_ResultSet = std::map<size_t, EOCR_Result_Single>;
using EOCRD_Rects = std::vector<std::vector<cv::Point2f>>;
}

@ -3,6 +3,9 @@
uns::OCRConfig uns::G_OCRConfig;
std::wstring detect_model_path;
std::wstring recognize_model_path;
uns::OCRConfig::OCRConfig()
{
language = easyocr::CharsetType::EN;

@ -25,8 +25,6 @@ namespace uns
private:
easyocr::GPUUsage gpu_usage;
easyocr::CharsetType language;
std::wstring detect_model_path;
std::wstring recognize_model_path;
public:
OCRConfig();

@ -11,8 +11,9 @@ namespace uns
using VecFloat = std::vector<float>;
using IONames = std::vector<const char*>;
using IONamesStorage = std::vector<std::string>;
using EOCR_Result = std::pair<std::wstring, float>;
using EOCR_ResultSet = std::map<size_t, EOCR_Result>;
using EOCR_Result = std::tuple<std::wstring, float>;
using EOCR_Result_Single = std::tuple<std::wstring, float, cv::Rect>;
using EOCR_ResultSet = std::map<size_t, EOCR_Result_Single>;
using EOCRD_Rects = std::vector<std::vector<cv::Point2f>>;
class OCRToolBox

Loading…
Cancel
Save