mirror of
https://github.com/RapidAI/RapidOCR.git
synced 2026-11-11 05:20:49 +08:00
134 lines
2.7 KiB
C++
Executable File
134 lines
2.7 KiB
C++
Executable File
#include "../include/precomp.h"
|
|
#ifdef __cplusplus
|
|
extern "C"
|
|
{
|
|
#endif
|
|
|
|
typedef struct
|
|
{
|
|
OcrLite OcrObj;
|
|
std::string strRes;
|
|
}OCR_OBJ;
|
|
|
|
_QM_OCR_API BPHANDLE BPOcrInit(const char* szDetModel, const char* szClsModel, const char* szRecModel,const char *szKeyPath, int nThreads)
|
|
{
|
|
|
|
omp_set_num_threads(nThreads);
|
|
OCR_OBJ* pOcrObj=new OCR_OBJ;
|
|
if(pOcrObj)
|
|
{
|
|
pOcrObj->OcrObj.setNumThread(nThreads);
|
|
|
|
pOcrObj->OcrObj.initModels(szDetModel, szClsModel, szRecModel, szKeyPath);
|
|
|
|
return pOcrObj;
|
|
}
|
|
else
|
|
{
|
|
return nullptr;
|
|
}
|
|
|
|
}
|
|
|
|
|
|
_QM_OCR_API BOOL BPOcrDoOcr(BPHANDLE handle, const char* szImgPath, BOOL bAutoParam,BOOL bLongPic,BPOCR_PARAM *pParam)
|
|
{
|
|
|
|
OCR_OBJ* pOcrObj=(OCR_OBJ*)handle;
|
|
if(!pOcrObj)
|
|
return FALSE;
|
|
|
|
BPOCR_PARAM Param = *pParam;
|
|
if(bAutoParam)
|
|
{
|
|
|
|
}
|
|
else
|
|
{
|
|
|
|
// -numThread %NUMBER_OF_PROCESSORS% ^
|
|
// --padding 0 ^
|
|
// --maxSideLen %MAX_SIDE_LEN% ^
|
|
// --boxScoreThresh 0.5 ^
|
|
// --boxThresh 0.3 ^
|
|
// --unClipRatio %CLIP_RATE% ^
|
|
// --doAngle 1 ^
|
|
// --mostAngle 0
|
|
if(Param.boxScoreThresh ==0)
|
|
Param.boxScoreThresh = 0.5;
|
|
if(Param.boxThresh ==0)
|
|
Param.boxThresh =0.3;
|
|
if(Param.flagDoAngle ==0)
|
|
Param.flagDoAngle =1;
|
|
if(Param.flagMostAngle ==0)
|
|
Param.flagMostAngle =0;
|
|
|
|
if( bLongPic)
|
|
{
|
|
if(Param.maxSideLen ==0)
|
|
Param.maxSideLen =0;
|
|
if(Param.unClipRatio ==0)
|
|
Param.unClipRatio = 3.0;
|
|
}
|
|
else
|
|
{
|
|
if(Param.maxSideLen ==0)
|
|
Param.maxSideLen =1024;
|
|
if(Param.unClipRatio ==0)
|
|
Param.unClipRatio = 1.5;
|
|
}
|
|
}
|
|
std::string imgPath,imgDir,imgName;
|
|
imgPath=szImgPath;
|
|
imgDir.assign(imgPath.substr(0, imgPath.find_last_of('\\') + 1));
|
|
imgName.assign(imgPath.substr(imgPath.find_last_of('\\') + 1));
|
|
OcrResult result=pOcrObj->OcrObj.detect(imgDir.c_str(),imgName.c_str(),Param.padding,Param.maxSideLen,Param.boxScoreThresh,Param.boxThresh,Param.unClipRatio,Param.flagDoAngle?true:false,Param.flagMostAngle?true:false);
|
|
if (result.strRes.length() > 0)
|
|
{
|
|
|
|
pOcrObj->strRes = result.strRes;
|
|
return TRUE;
|
|
}
|
|
else
|
|
return FALSE;
|
|
}
|
|
|
|
|
|
_QM_OCR_API int BPOcrGetLen(BPHANDLE handle)
|
|
{
|
|
OCR_OBJ* pOcrObj = (OCR_OBJ*)handle;
|
|
if (!pOcrObj)
|
|
return 0;
|
|
return pOcrObj->strRes.size()+1;
|
|
}
|
|
|
|
_QM_OCR_API BOOL BPOcrGetResult(BPHANDLE handle, char* szBuf, int nLen)
|
|
{
|
|
|
|
OCR_OBJ* pOcrObj = (OCR_OBJ*)handle;
|
|
if (!pOcrObj)
|
|
return FALSE;
|
|
|
|
if (nLen > pOcrObj->strRes.size())
|
|
{
|
|
strncpy(szBuf, pOcrObj->strRes.c_str(), pOcrObj->strRes.size());
|
|
szBuf[pOcrObj->strRes.size() - 1] = 0;
|
|
}
|
|
|
|
return pOcrObj->strRes.size();
|
|
}
|
|
|
|
_QM_OCR_API void BPOcrDeinit(BPHANDLE handle)
|
|
{
|
|
|
|
OCR_OBJ* pOcrObj=(OCR_OBJ*)handle;
|
|
if(pOcrObj)
|
|
delete pOcrObj;
|
|
|
|
|
|
}
|
|
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif |