barcodeDetector.h
1.4 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
#pragma once
//
// yoloWrapper¡¤±êÍ·
//
#ifndef __BARCODEDETECTOR_H
#define __BARCODEDETECTOR_H
#include <opencv2\opencv.hpp>
class BarcodeDetector
{
public:
BarcodeDetector() {};
~BarcodeDetector() {};
bool detect(cv::Mat img, std::vector<cv::Point>& points);
void drawBarcode(cv::Mat& img, const std::vector<cv::Point>& points);
private:
std::vector<cv::RotatedRect> localization_rects;
std::vector<cv::RotatedRect> localization_bbox;
std::vector<float> bbox_scores;
std::vector<int> bbox_indices;
std::vector<std::vector<cv::Point2f>> transformation_points;
bool checkBarInputImage(cv::Mat img, cv::Mat& gray);
void updatePointsResult(cv::OutputArray points_, const std::vector<cv::Point2f> &points);
void init(const cv::Mat& src);
void localization();
std::vector<std::vector<cv::Point2f>> getTransformationPoints() {
return transformation_points;
}
bool computeTransformationPoints();
protected:
enum resize_direction
{
ZOOMING, SHRINKING, UNCHANGED
}purpose = UNCHANGED;
double coeff = 1.0;
int height, width;
cv::Mat resized_barcode, gradient_magnitude, coherence, orientation, edge_nums, integral_x_sq, integral_y_sq, integral_xy, integral_edges;
void preprocess();
void calCoherence(int window_size);
static inline bool isValidCoord(const cv::Point &coord, const cv::Size &limit);
void regionGrowing(int window_size);
void barcodeErode();
};
#endif/* __BARCODEDETECTOR_H */