Public-Mirror/LunaTranslator
1
0
Fork 0
mirror of https://github.com/HIllya51/LunaTranslator.git synced 2024-12-27 15:44:12 +08:00
Code Issues Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
恍兮惚兮 2024-11-15 03:44:19 +08:00
parent fcb05dbcc4
commit ec1e74ce41
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
cpp/LunaOCR/OCR.cpp
View File

@ -1,3 +1,4 @@
#ifndef WINXP
#include <onnxruntime/core/session/onnxruntime_cxx_api.h> #include <onnxruntime/core/session/onnxruntime_cxx_api.h>
#include <opencv2/opencv.hpp> #include <opencv2/opencv.hpp>
#include <opencv2/imgproc.hpp> #include <opencv2/imgproc.hpp>
@ -632,13 +633,13 @@ std::vector<cv::Mat> OcrLite::getPartImages(cv::Mat &src, std::vector<TextBox> &
return partImages; return partImages;
} }
void matRotateClockWise180(cv::Mat& src) void matRotateClockWise180(cv::Mat &src)
{ {
flip(src, src, 0); flip(src, src, 0);
flip(src, src, 1); flip(src, src, 1);
} }
void matRotateClockWise90(cv::Mat& src) void matRotateClockWise90(cv::Mat &src)
{ {
transpose(src, src); transpose(src, src);
flip(src, src, 1); flip(src, src, 1);
@ -704,13 +705,14 @@ std::vector<TextBlock> OcrLite::detect_internal(cv::Mat &src, cv::Rect &originRe
return textBlocks; return textBlocks;
} }
#endif
struct ocrpoints struct ocrpoints
{ {
int x1, y1, x2, y2, x3, y3, x4, y4; int x1, y1, x2, y2, x3, y3, x4, y4;
}; };
DECLARE_API OcrLite *OcrInit(const wchar_t *szDetModel, const wchar_t *szRecModel, const wchar_t *szKeyPath, int nThreads) DECLARE_API OcrLite *OcrInit(const wchar_t *szDetModel, const wchar_t *szRecModel, const wchar_t *szKeyPath, int nThreads)
{ {
#ifndef WINXP
OcrLite *pOcrObj = nullptr; OcrLite *pOcrObj = nullptr;
try try
{ {
@ -727,10 +729,14 @@ DECLARE_API OcrLite *OcrInit(const wchar_t *szDetModel, const wchar_t *szRecMode
{ {
return nullptr; return nullptr;
} }
#else
return nullptr;
#endif
} }
DECLARE_API void OcrDetect(OcrLite *pOcrObj, const void *binptr, size_t size, Directional mode, void (*cb)(ocrpoints, const char *)) DECLARE_API void OcrDetect(OcrLite *pOcrObj, const void *binptr, size_t size, Directional mode, void (*cb)(ocrpoints, const char *))
{ {
#ifndef WINXP
if (!pOcrObj) if (!pOcrObj)
return; return;
@ -750,10 +756,13 @@ DECLARE_API void OcrDetect(OcrLite *pOcrObj, const void *binptr, size_t size, Di
catch (...) catch (...)
{ {
} }
#endif
} }
DECLARE_API void OcrDestroy(OcrLite *pOcrObj) DECLARE_API void OcrDestroy(OcrLite *pOcrObj)
{ {
#ifndef WINXP
if (pOcrObj) if (pOcrObj)
delete pOcrObj; delete pOcrObj;
#endif
} }