80 lines
2.4 KiB
C++
80 lines
2.4 KiB
C++
#pragma once
|
||
|
||
#include <vector>
|
||
#include "opencv2/opencv.hpp"
|
||
|
||
namespace cvpr
|
||
{
|
||
enum class SMOOTH_METHOD {
|
||
NONE = 0,
|
||
BSPLINE
|
||
//DEANGEL
|
||
};
|
||
|
||
class __declspec(dllexport) IContoursExtractor {
|
||
public:
|
||
virtual ~IContoursExtractor() = default;
|
||
|
||
virtual void init() = 0;
|
||
|
||
virtual void destroy() = 0;
|
||
|
||
/*
|
||
* threshold, 轮廓检测结果中小于该阈值的轮廓将被剔除
|
||
*/
|
||
virtual void setAreaThreshold(double threshold = 500.0) = 0;
|
||
|
||
virtual double areaThreshold() = 0;
|
||
|
||
/*
|
||
* contourOffset, 轮廓的偏置距离（单位：像素），当contourOffset取值为0时，该函数自动忽略setSmoothMethod()的method参数，不做平滑直接返回原始轮廓
|
||
*/
|
||
virtual void setContourOffset(int32_t offset = 12) = 0;
|
||
|
||
virtual int32_t contourOffset() = 0;
|
||
|
||
/*
|
||
* epsilon, 原始轮廓偏置后的轮廓线的多边形逼近精度
|
||
*/
|
||
virtual void setApproxPolyEpsilon(double epsilon = 5) = 0;
|
||
|
||
virtual double approxPolyEpsilon() = 0;
|
||
|
||
/*
|
||
* method, 原始轮廓偏置后的轮廓线的平滑方法，默认采用b样条曲线
|
||
*/
|
||
virtual void setSmoothMethod(SMOOTH_METHOD method = SMOOTH_METHOD::BSPLINE) = 0;
|
||
|
||
virtual SMOOTH_METHOD smoothMethod() = 0;
|
||
|
||
/*
|
||
* threshold, 采用多边形逼近之后，如果折线的夹角小于该阈值，则采用b样条曲线进行平滑。该参数仅在SMOOTH_METHOD::DEANGEL模式下有效
|
||
*/
|
||
virtual void setAngelThreshold(double threshold = 20.0) = 0;
|
||
|
||
virtual double angelThreshold() = 0;
|
||
|
||
/*
|
||
* threshold, 采用多边形逼近之后，如果缺陷处的开口小于该阈值，则封闭该缺陷
|
||
*/
|
||
virtual void setDefectThreshold(double threshold = 0.0) = 0;
|
||
|
||
virtual double defectThreshold() = 0;
|
||
|
||
/*
|
||
* smoothStep, 原始轮廓偏置后的轮廓线的平滑步长，取值范围（0.0，1.0）
|
||
*/
|
||
virtual void setSmoothStep(double step = 0.001) = 0;
|
||
|
||
virtual double smoothStep() = 0;
|
||
|
||
/* 描述: 提取轮廓或轮廓偏置线
|
||
* 参数：
|
||
* alpha, 图像alpha通道
|
||
* outputContours, 输出轮廓集合
|
||
*/
|
||
virtual void extract(const cv::Mat& alpha, std::vector<std::vector<cv::Point>>& outputContours) = 0;
|
||
|
||
};
|
||
}
|