116 lines
4.4 KiB
C++
116 lines
4.4 KiB
C++
/*
|
||
* ====================================================
|
||
|
||
* 功能：Mark点定位裁剪。识别试卷中的mark点，进行纠偏裁剪。该功能不能完全独立使用，因为可能会识别失败，然后需要继续用传统裁剪纠偏进行补充。
|
||
* 作者：刘丁维
|
||
* 生成时间：2020/5/22
|
||
* 最近修改时间：2020/5/22 v1.0
|
||
2020/7/29 v1.1 优化Mark点识别，以及文稿方向判定
|
||
2020/7/30 v1.1.1 优化Mark点识别，以及文稿方向判定
|
||
2020/8/3 v1.2 针对七天，添加三Mark点识别文稿方向，原来的方式定义为多边识别文稿方向
|
||
2020/8/12 v1.3 针对七天，添加了条码判断算法，用于判定试卷是否为指定试卷。该条码只是粗略识别是否是条码，而不进行准确识别。
|
||
apply接口增加barCode参数，用于是否开启粗判断条码功能
|
||
2020/8/12 v1.4 针对七天，添加裁剪接口，可进行裁剪与不裁剪两种选择
|
||
* 版本号：v1.4
|
||
|
||
* ====================================================
|
||
*/
|
||
|
||
#ifndef IMAGEMARKCROPAPPLY_H
|
||
#define IMAGEMARKCROPAPPLY_H
|
||
|
||
#include "ImageApply.h"
|
||
|
||
class GIMGPROC_LIBRARY_API CImageApplyMarkCrop
|
||
{
|
||
public:
|
||
|
||
//识别文稿方向方式
|
||
enum DirectionMethod
|
||
{
|
||
Trilateral_7Net, //三边定位文稿方向（针对七天需求）
|
||
Multilateral //多边定位文稿方向
|
||
};
|
||
|
||
enum DPI : ushort
|
||
{
|
||
DPI_200 = 200,
|
||
DPI_300 = 300
|
||
};
|
||
|
||
enum DeviceModel
|
||
{
|
||
G400 = 2,
|
||
G300 = 3
|
||
};
|
||
|
||
enum Orientation
|
||
{
|
||
Default = 0,
|
||
Left_RT, //多边文稿方向识别时靠左为正；7天3mark识别时，缺角在右上为正
|
||
Top_RB, //多边文稿方向识别时靠上为正；7天3mark识别时，缺角在右下为正
|
||
Right_LB, //多边文稿方向识别时靠右为正；7天3mark识别时，缺角在左下为正
|
||
Bottom_LT //多边文稿方向识别时靠下为正；7天3mark识别时，缺角在左上为正
|
||
};
|
||
|
||
public:
|
||
CImageApplyMarkCrop(DeviceModel device, bool isCropped = true, double threshold = 20, int noise = 40, DPI dpi = DPI::DPI_200, DirectionMethod direction = DirectionMethod::Multilateral);
|
||
|
||
/*
|
||
* src [in]: 原图
|
||
* dst [out]:处理成功时，返回的结果图像
|
||
* markOri [in]: 文稿方向判定方式
|
||
* barCode [in]: 是否判断条形码，该功能只针对Trilateral_7Net模式
|
||
* angle [out]:返回文稿方向偏转角度。有0、90、180、270四种情况
|
||
* 返回值： 0：成功；
|
||
-1：原图为空，或者无法找到图片轮廓（纯黑图会报该错误）；
|
||
-2：mark小于3个
|
||
-3：选择Multilateral模式时，mark点小于4个
|
||
-4：识别条形码时，目标图像过小，无法包含条形码
|
||
-5：无条形码
|
||
*/
|
||
int apply(const cv::Mat& src, cv::Mat& dst, Orientation markOri, bool barCode, int& angle);
|
||
|
||
inline DeviceModel getDeviceModel() { return m_device; }
|
||
|
||
inline void setDeviceModel(DeviceModel device) { m_device = device; }
|
||
|
||
inline bool isCropped() { return m_b_isCropped; }
|
||
|
||
inline void setCropped(bool cropped) { m_b_isCropped = cropped; }
|
||
|
||
inline double getThreshold() { return m_threshold; }
|
||
|
||
inline void setThreshold(double threshold) { m_threshold = threshold; }
|
||
|
||
inline int getNoise() { return m_noise; }
|
||
|
||
inline void setNoise(int noise) { m_noise = noise; }
|
||
|
||
inline DPI getDPI() { return m_dpi; }
|
||
|
||
inline void setDPI(DPI dpi) { m_dpi = dpi; }
|
||
|
||
inline const cv::Range getSizeRange() { return m_range; }
|
||
|
||
inline void setSizeRange(const cv::Range& range) { m_range = range; }
|
||
|
||
inline void setSizeRange(int low, int up) { m_range = cv::Range(low, up); }
|
||
|
||
private:
|
||
|
||
void findMarks(const std::vector<std::vector<cv::Point>>& contours, const cv::RotatedRect& region, const cv::Range& range,
|
||
std::vector<std::vector<cv::Point>>& marks, std::vector<cv::RotatedRect>& rrect);
|
||
|
||
bool isContainBarCode(const cv::Mat& image);
|
||
private:
|
||
DeviceModel m_device;
|
||
bool m_b_isCropped;
|
||
double m_threshold;
|
||
int m_noise;
|
||
DPI m_dpi;
|
||
cv::Range m_range;
|
||
DirectionMethod m_direction;
|
||
};
|
||
|
||
#endif // IMAGEMARKCROPAPPLY_H
|