44 lines
1.4 KiB
C
44 lines
1.4 KiB
C
|
#ifndef DETECTIONPROCESSING_H
|
||
|
|
#define DETECTIONPROCESSING_H
|
||
|
|
|
||
|
|
#include <QObject>
|
||
|
|
#include "processing.h"
|
||
|
|
#include <memory>
|
||
|
|
|
||
|
|
class DetectionProcessing : public Processing {
|
||
|
|
|
||
|
|
Q_OBJECT
|
||
|
|
|
||
|
|
public:
|
||
|
|
DetectionProcessing(QObject *parent = 0);
|
||
|
|
void init(Json::Value json_steps);
|
||
|
|
const cv::Mat& getResult();
|
||
|
|
const cv::Mat& getMarkers();
|
||
|
|
int getNutCount();
|
||
|
|
|
||
|
|
// Processing methods
|
||
|
|
public slots:
|
||
|
|
cv::Mat _dilate(const cv::Mat& input , int kernel_size, int iterations);
|
||
|
|
cv::Mat _threshold(const cv::Mat& input, int ignored, int ignored2);
|
||
|
|
cv::Mat _thresh_color(const cv::Mat& input, int ignored, int ignored2);
|
||
|
|
cv::Mat _erode(const cv::Mat& input , int kernel_size, int iterations);
|
||
|
|
cv::Mat _erode_intermediate(const cv::Mat& input , int kernel_size, int iterations);
|
||
|
|
cv::Mat _watershed(const cv::Mat& input, int ignored, int ignored2);
|
||
|
|
cv::Mat _invert(const cv::Mat& input, int ignored, int ignored2);
|
||
|
|
cv::Mat _invert_markers(const cv::Mat& input, int ignored, int ignored2);
|
||
|
|
cv::Mat _resize(const cv::Mat& input, int new_length, int ignored);
|
||
|
|
cv::Mat _resizeToOriginal(const cv::Mat& input, int ignored, int ignored2);
|
||
|
|
cv::Mat _markBackground(const cv::Mat& input, int ignored, int ignored2);
|
||
|
|
|
||
|
|
private:
|
||
|
|
const cv::Mat* getInput();
|
||
|
|
|
||
|
|
// Members
|
||
|
|
private:
|
||
|
|
cv::Mat markers;
|
||
|
|
int nut_count;
|
||
|
|
cv::Mat intermediate;
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif // DETECTIONPROCESSING_H
|