You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
136 lines
3.1 KiB
136 lines
3.1 KiB
#ifndef EASYPR_CORE_PLATEDETECT_H_
|
|
#define EASYPR_CORE_PLATEDETECT_H_
|
|
|
|
#include "easypr/core/plate_locate.h"
|
|
#include "easypr/core/plate_judge.h"
|
|
|
|
namespace easypr {
|
|
|
|
class CPlateDetect {
|
|
public:
|
|
CPlateDetect();
|
|
|
|
~CPlateDetect();
|
|
|
|
/** @brief Plate detect in an image.
|
|
|
|
The function detects plate in an image. It can use sobel, color, and character method or the combinations of them.
|
|
|
|
@param src Source image.
|
|
@param resultVec Destination vector of CPlate.
|
|
@param type Detect type. (eg. PR_DETECT_SOBEL + PR_DETECT_COLOR)
|
|
@param showDetectArea
|
|
@param index
|
|
*/
|
|
int plateDetect(Mat src, std::vector<CPlate> &resultVec, int type,
|
|
bool showDetectArea = true, int index = 0);
|
|
|
|
|
|
/** @brief Plate detect in an image.
|
|
|
|
The function detects plate in an image. It can use sobel, color, and character method or the combinations of them.
|
|
Use default m_type, it can use setDetectType() to set it;
|
|
|
|
@param src Source image.
|
|
@param resultVec Destination vector of CPlate.
|
|
@param showDetectArea
|
|
@param index
|
|
*/
|
|
int plateDetect(Mat src, std::vector<CPlate> &resultVec,
|
|
bool showDetectArea = true, int index = 0);
|
|
|
|
|
|
//! 展示中间的结果
|
|
|
|
int showResult(const Mat &result);
|
|
|
|
//! 装载SVM模型
|
|
|
|
void LoadSVM(std::string s);
|
|
|
|
//! 生活模式与工业模式切换
|
|
|
|
inline void setPDLifemode(bool param) { m_plateLocate->setLifemode(param); }
|
|
|
|
//! 是否开启调试模式
|
|
|
|
inline void setPDDebug(bool param) { m_plateLocate->setDebug(param); }
|
|
|
|
//! 获取调试模式状态
|
|
|
|
inline bool getPDDebug() { return m_plateLocate->getDebug(); }
|
|
|
|
|
|
inline void setDetectType(int param) { m_type = param; }
|
|
|
|
//! 设置与读取变量
|
|
|
|
inline void setGaussianBlurSize(int param) {
|
|
m_plateLocate->setGaussianBlurSize(param);
|
|
}
|
|
|
|
inline int getGaussianBlurSize() const {
|
|
return m_plateLocate->getGaussianBlurSize();
|
|
}
|
|
|
|
inline void setMorphSizeWidth(int param) {
|
|
m_plateLocate->setMorphSizeWidth(param);
|
|
}
|
|
|
|
inline int getMorphSizeWidth() const {
|
|
return m_plateLocate->getMorphSizeWidth();
|
|
}
|
|
|
|
inline void setMorphSizeHeight(int param) {
|
|
m_plateLocate->setMorphSizeHeight(param);
|
|
}
|
|
|
|
inline int getMorphSizeHeight() const {
|
|
return m_plateLocate->getMorphSizeHeight();
|
|
}
|
|
|
|
inline void setVerifyError(float param) {
|
|
m_plateLocate->setVerifyError(param);
|
|
}
|
|
|
|
inline float getVerifyError() const {
|
|
return m_plateLocate->getVerifyError();
|
|
}
|
|
|
|
inline void setVerifyAspect(float param) {
|
|
m_plateLocate->setVerifyAspect(param);
|
|
}
|
|
|
|
inline float getVerifyAspect() const {
|
|
return m_plateLocate->getVerifyAspect();
|
|
}
|
|
|
|
inline void setVerifyMin(int param) { m_plateLocate->setVerifyMin(param); }
|
|
|
|
inline void setVerifyMax(int param) { m_plateLocate->setVerifyMax(param); }
|
|
|
|
inline void setJudgeAngle(int param) { m_plateLocate->setJudgeAngle(param); }
|
|
|
|
inline void setMaxPlates(int param) { m_maxPlates = param; }
|
|
|
|
inline int getMaxPlates() const { return m_maxPlates; }
|
|
|
|
private:
|
|
|
|
//! 设置一幅图中最多有多少车牌
|
|
|
|
int m_maxPlates;
|
|
|
|
//! 车牌定位
|
|
|
|
CPlateLocate* m_plateLocate;
|
|
|
|
int m_type;
|
|
|
|
static std::string m_pathSvm;
|
|
|
|
};
|
|
|
|
}
|
|
|
|
#endif // EASYPR_CORE_PLATEDETECT_H_
|