You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
78 lines
1.5 KiB
C
78 lines
1.5 KiB
C
5 years ago
|
#ifndef _EDColor_
|
||
|
|
#define _EDColor_
|
||
|
|
|
||
|
|
#include <opencv2/opencv.hpp>
|
||
|
|
|
||
|
|
// Look up table size for fast color space conversion
|
||
|
|
#define LUT_SIZE (1024*4096)
|
||
|
|
|
||
|
|
// Special defines
|
||
|
|
#define EDGE_VERTICAL 1
|
||
|
|
#define EDGE_HORIZONTAL 2
|
||
|
|
#define EDGE_45 3
|
||
|
|
#define EDGE_135 4
|
||
|
|
|
||
|
|
#define MAX_GRAD_VALUE 128*256
|
||
|
|
#define EPSILON 1.0
|
||
|
|
#define MIN_PATH_LEN 10
|
||
|
|
|
||
|
|
|
||
|
|
class EDColor {
|
||
|
|
public:
|
||
|
|
EDColor(cv::Mat srcImage, int gradThresh = 20, int anchor_thresh = 4, double sigma = 1.5, bool validateSegments=false);
|
||
|
|
cv::Mat getEdgeImage();
|
||
|
|
std::vector<std::vector<cv::Point>> getSegments();
|
||
|
|
int getSegmentNo();
|
||
|
|
|
||
|
|
int getWidth();
|
||
|
|
int getHeight();
|
||
|
|
|
||
|
|
cv::Mat inputImage;
|
||
|
|
private:
|
||
|
|
uchar *L_Img;
|
||
|
|
uchar *a_Img;
|
||
|
|
uchar *b_Img;
|
||
|
|
|
||
|
|
uchar *smooth_L;
|
||
|
|
uchar *smooth_a;
|
||
|
|
uchar *smooth_b;
|
||
|
|
|
||
|
|
uchar *dirImg;
|
||
|
|
short *gradImg;
|
||
|
|
|
||
|
|
cv::Mat edgeImage;
|
||
|
|
uchar *edgeImg;
|
||
|
|
|
||
|
|
const uchar *blueImg;
|
||
|
|
const uchar *greenImg;
|
||
|
|
const uchar *redImg;
|
||
|
|
|
||
|
|
int width;
|
||
|
|
int height;
|
||
|
|
|
||
|
|
double divForTestSegment;
|
||
|
|
double *H;
|
||
|
|
int np;
|
||
|
|
int segmentNo;
|
||
|
|
|
||
|
|
std::vector<std::vector<cv::Point>> segments;
|
||
|
|
|
||
|
|
static double LUT1[LUT_SIZE + 1];
|
||
|
|
static double LUT2[LUT_SIZE + 1];
|
||
|
|
static bool LUT_Initialized;
|
||
|
|
|
||
|
|
void MyRGB2LabFast();
|
||
|
|
void ComputeGradientMapByDiZenzo();
|
||
|
|
void smoothChannel(uchar *src, uchar *smooth, double sigma);
|
||
|
|
void validateEdgeSegments();
|
||
|
|
void testSegment(int i, int index1, int index2);
|
||
|
|
void extractNewSegments();
|
||
|
|
double NFA(double prob, int len);
|
||
|
|
|
||
|
|
static void fixEdgeSegments(std::vector<std::vector<cv::Point>> map, int noPixels);
|
||
|
|
|
||
|
|
static void InitColorEDLib();
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif // ! _EDColor_
|