You've already forked UnitV2Framework
mirror of
https://github.com/m5stack/UnitV2Framework.git
synced 2026-05-20 10:30:55 -07:00
icyqwq
40 lines
919 B
C++
40 lines
919 B
C++
#ifndef __OCR_CRNNNET_H__
|
|
#define __OCR_CRNNNET_H__
|
|
|
|
#include "OcrStruct.h"
|
|
#include "ncnn/net.h"
|
|
#include <opencv2/core/core.hpp>
|
|
#include <opencv2/imgproc/imgproc.hpp>
|
|
|
|
class CrnnNet {
|
|
public:
|
|
|
|
~CrnnNet();
|
|
|
|
void setNumThread(int numOfThread);
|
|
|
|
void setGpuIndex(int gpuIndex);
|
|
|
|
bool initModel(const std::string &pathStr, const std::string &keysPath);
|
|
|
|
std::vector<TextLine> getTextLines(std::vector<cv::Mat> &partImg, const char *path, const char *imgName);
|
|
|
|
private:
|
|
bool isOutputDebugImg = false;
|
|
int numThread;
|
|
ncnn::Net net;
|
|
|
|
const float meanValues[3] = {127.5, 127.5, 127.5};
|
|
const float normValues[3] = {1.0 / 127.5, 1.0 / 127.5, 1.0 / 127.5};
|
|
const int dstHeight = 32;
|
|
|
|
std::vector<std::string> keys;
|
|
|
|
TextLine scoreToTextLine(const std::vector<float> &outputData, int h, int w);
|
|
|
|
TextLine getTextLine(const cv::Mat &src);
|
|
};
|
|
|
|
|
|
#endif //__OCR_CRNNNET_H__
|