bob.pan
5 years ago
parent
77ea01979b
commit
592c491ac2
@ -0,0 +1,94 @@
|
|||||||
|
#include "AlgoTest.h"
|
||||||
|
#include <QGridLayout>
|
||||||
|
#include <QDir>
|
||||||
|
#include <QFileInfo>
|
||||||
|
#include <QDirIterator>
|
||||||
|
|
||||||
|
#pragma execution_character_set("utf-8")
|
||||||
|
|
||||||
|
AlgoTest::AlgoTest(QWidget *parent)
|
||||||
|
: QMainWindow(parent)
|
||||||
|
{
|
||||||
|
ui.setupUi(this);
|
||||||
|
connect(ui.pushButton, SIGNAL(clicked()), this, SLOT(onButtonClicked()));
|
||||||
|
|
||||||
|
{
|
||||||
|
QGridLayout *pLayout = new QGridLayout(ui.widget_src);
|
||||||
|
m_srcView = new lpImgViewer(ui.widget_src);
|
||||||
|
pLayout->addWidget(m_srcView);
|
||||||
|
ui.widget_src->setLayout(pLayout);
|
||||||
|
}
|
||||||
|
{
|
||||||
|
QGridLayout *pLayout = new QGridLayout(ui.widget_rlt);
|
||||||
|
m_rltView = new lpImgViewer(ui.widget_rlt);
|
||||||
|
pLayout->addWidget(m_rltView);
|
||||||
|
ui.widget_rlt->setLayout(pLayout);
|
||||||
|
}
|
||||||
|
loadImage();
|
||||||
|
|
||||||
|
m_pAlg = new QAlgDetect();
|
||||||
|
}
|
||||||
|
|
||||||
|
AlgoTest::~AlgoTest()
|
||||||
|
{
|
||||||
|
if (m_rltView)
|
||||||
|
{
|
||||||
|
delete m_rltView;
|
||||||
|
m_rltView = nullptr;
|
||||||
|
}
|
||||||
|
if (m_srcView) {
|
||||||
|
delete m_srcView;
|
||||||
|
m_srcView = nullptr;
|
||||||
|
}
|
||||||
|
if (m_pAlg)
|
||||||
|
{
|
||||||
|
delete m_pAlg;
|
||||||
|
m_pAlg = nullptr;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void AlgoTest::loadImage()
|
||||||
|
{
|
||||||
|
QString strPath = "H:\\н¨Îļþ¼Ð\\Source\\";
|
||||||
|
//QString strPath = "H:\\н¨Îļþ¼Ð\\7\\";
|
||||||
|
QStringList fileFilter;
|
||||||
|
fileFilter<< "*.BMP" << "*.JPEG" << "*.JPG" << "*.PNG";
|
||||||
|
|
||||||
|
QDirIterator qdi(strPath,
|
||||||
|
fileFilter,
|
||||||
|
QDir::NoSymLinks | QDir::Files,
|
||||||
|
QDirIterator::Subdirectories);
|
||||||
|
while (qdi.hasNext())
|
||||||
|
{
|
||||||
|
m_filelst.append(qdi.next());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void AlgoTest::algCallFunc(QImage img)
|
||||||
|
{
|
||||||
|
m_rltView->setImg(img);
|
||||||
|
}
|
||||||
|
|
||||||
|
Q_SLOT void AlgoTest::onButtonClicked()
|
||||||
|
{
|
||||||
|
QString strObj = sender()->objectName();
|
||||||
|
if (strObj == "pushButton")
|
||||||
|
{
|
||||||
|
if (m_filelst.size() <= 0)
|
||||||
|
return;
|
||||||
|
if (m_gIndex >= m_filelst.size())
|
||||||
|
m_gIndex = 0;
|
||||||
|
QString strPath = m_filelst.at(m_gIndex);
|
||||||
|
m_gIndex++;
|
||||||
|
|
||||||
|
QImage img;
|
||||||
|
bool b = img.load(strPath);
|
||||||
|
if (b == true)
|
||||||
|
{
|
||||||
|
m_srcView->setImg(img);
|
||||||
|
}
|
||||||
|
|
||||||
|
AlgCallBack func = std::bind(&AlgoTest::algCallFunc, this, std::placeholders::_1);
|
||||||
|
m_pAlg->detect(img, func);
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,32 @@
|
|||||||
|
#ifndef _ALGOTEST_H_
|
||||||
|
#define _ALGOTEST_H_
|
||||||
|
|
||||||
|
#include <QtWidgets/QMainWindow>
|
||||||
|
#include "ui_AlgoTest.h"
|
||||||
|
#include "lpImgViewer.h"
|
||||||
|
#include "QAlgDetect.h"
|
||||||
|
|
||||||
|
class AlgoTest : public QMainWindow
|
||||||
|
{
|
||||||
|
Q_OBJECT
|
||||||
|
|
||||||
|
public:
|
||||||
|
AlgoTest(QWidget *parent = Q_NULLPTR);
|
||||||
|
~AlgoTest();
|
||||||
|
void loadImage();
|
||||||
|
void algCallFunc(QImage img);
|
||||||
|
Q_SLOT void onButtonClicked();
|
||||||
|
private:
|
||||||
|
Ui::AlgoTestClass ui;
|
||||||
|
|
||||||
|
lpImgViewer* m_srcView{ nullptr };
|
||||||
|
lpImgViewer* m_rltView{ nullptr };
|
||||||
|
|
||||||
|
private:
|
||||||
|
QStringList m_filelst;
|
||||||
|
int m_gIndex{ 0 };
|
||||||
|
|
||||||
|
QAlgDetect *m_pAlg{ nullptr };
|
||||||
|
};
|
||||||
|
#endif
|
||||||
|
|
||||||
@ -0,0 +1,4 @@
|
|||||||
|
<RCC>
|
||||||
|
<qresource prefix="AlgoTest">
|
||||||
|
</qresource>
|
||||||
|
</RCC>
|
||||||
@ -0,0 +1,90 @@
|
|||||||
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
|
<ui version="4.0">
|
||||||
|
<class>AlgoTestClass</class>
|
||||||
|
<widget class="QMainWindow" name="AlgoTestClass">
|
||||||
|
<property name="geometry">
|
||||||
|
<rect>
|
||||||
|
<x>0</x>
|
||||||
|
<y>0</y>
|
||||||
|
<width>681</width>
|
||||||
|
<height>748</height>
|
||||||
|
</rect>
|
||||||
|
</property>
|
||||||
|
<property name="windowTitle">
|
||||||
|
<string>AlgoTest</string>
|
||||||
|
</property>
|
||||||
|
<widget class="QWidget" name="centralWidget">
|
||||||
|
<layout class="QGridLayout" name="gridLayout_4">
|
||||||
|
<item row="0" column="1">
|
||||||
|
<widget class="QPushButton" name="pushButton">
|
||||||
|
<property name="text">
|
||||||
|
<string>PushButton</string>
|
||||||
|
</property>
|
||||||
|
</widget>
|
||||||
|
</item>
|
||||||
|
<item row="1" column="1">
|
||||||
|
<spacer name="verticalSpacer">
|
||||||
|
<property name="orientation">
|
||||||
|
<enum>Qt::Vertical</enum>
|
||||||
|
</property>
|
||||||
|
<property name="sizeHint" stdset="0">
|
||||||
|
<size>
|
||||||
|
<width>20</width>
|
||||||
|
<height>40</height>
|
||||||
|
</size>
|
||||||
|
</property>
|
||||||
|
</spacer>
|
||||||
|
</item>
|
||||||
|
<item row="0" column="0" rowspan="2">
|
||||||
|
<widget class="QWidget" name="widget" native="true">
|
||||||
|
<property name="sizePolicy">
|
||||||
|
<sizepolicy hsizetype="Expanding" vsizetype="Expanding">
|
||||||
|
<horstretch>0</horstretch>
|
||||||
|
<verstretch>0</verstretch>
|
||||||
|
</sizepolicy>
|
||||||
|
</property>
|
||||||
|
<layout class="QGridLayout" name="gridLayout_3">
|
||||||
|
<item row="0" column="0">
|
||||||
|
<widget class="QGroupBox" name="groupBox">
|
||||||
|
<property name="title">
|
||||||
|
<string>原图</string>
|
||||||
|
</property>
|
||||||
|
<layout class="QGridLayout" name="gridLayout">
|
||||||
|
<item row="0" column="0">
|
||||||
|
<widget class="QWidget" name="widget_src" native="true">
|
||||||
|
<property name="styleSheet">
|
||||||
|
<string notr="true">background-color: rgb(255, 223, 208);</string>
|
||||||
|
</property>
|
||||||
|
</widget>
|
||||||
|
</item>
|
||||||
|
</layout>
|
||||||
|
</widget>
|
||||||
|
</item>
|
||||||
|
<item row="1" column="0">
|
||||||
|
<widget class="QGroupBox" name="groupBox_2">
|
||||||
|
<property name="title">
|
||||||
|
<string>结果</string>
|
||||||
|
</property>
|
||||||
|
<layout class="QGridLayout" name="gridLayout_2">
|
||||||
|
<item row="0" column="0">
|
||||||
|
<widget class="QWidget" name="widget_rlt" native="true">
|
||||||
|
<property name="styleSheet">
|
||||||
|
<string notr="true">background-color: rgb(194, 194, 194);</string>
|
||||||
|
</property>
|
||||||
|
</widget>
|
||||||
|
</item>
|
||||||
|
</layout>
|
||||||
|
</widget>
|
||||||
|
</item>
|
||||||
|
</layout>
|
||||||
|
</widget>
|
||||||
|
</item>
|
||||||
|
</layout>
|
||||||
|
</widget>
|
||||||
|
</widget>
|
||||||
|
<layoutdefault spacing="6" margin="11"/>
|
||||||
|
<resources>
|
||||||
|
<include location="AlgoTest.qrc"/>
|
||||||
|
</resources>
|
||||||
|
<connections/>
|
||||||
|
</ui>
|
||||||
@ -0,0 +1,67 @@
|
|||||||
|
<?xml version="1.0" encoding="utf-8"?>
|
||||||
|
<Project ToolsVersion="4.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
|
||||||
|
<ItemGroup>
|
||||||
|
<Filter Include="Source Files">
|
||||||
|
<UniqueIdentifier>{4FC737F1-C7A5-4376-A066-2A32D752A2FF}</UniqueIdentifier>
|
||||||
|
<Extensions>cpp;c;cc;cxx;def;odl;idl;hpj;bat;asm;asmx</Extensions>
|
||||||
|
</Filter>
|
||||||
|
<Filter Include="Header Files">
|
||||||
|
<UniqueIdentifier>{93995380-89BD-4b04-88EB-625FBE52EBFB}</UniqueIdentifier>
|
||||||
|
<Extensions>h;hh;hpp;hxx;hm;inl;inc;xsd</Extensions>
|
||||||
|
</Filter>
|
||||||
|
<Filter Include="Resource Files">
|
||||||
|
<UniqueIdentifier>{D9D6E242-F8AF-46E4-B9FD-80ECBC20BA3E}</UniqueIdentifier>
|
||||||
|
<Extensions>qrc;*</Extensions>
|
||||||
|
<ParseFiles>false</ParseFiles>
|
||||||
|
</Filter>
|
||||||
|
<Filter Include="Form Files">
|
||||||
|
<UniqueIdentifier>{99349809-55BA-4b9d-BF79-8FDBB0286EB3}</UniqueIdentifier>
|
||||||
|
<Extensions>ui</Extensions>
|
||||||
|
</Filter>
|
||||||
|
<Filter Include="Resource Files">
|
||||||
|
<UniqueIdentifier>{D9D6E242-F8AF-46E4-B9FD-80ECBC20BA3E}</UniqueIdentifier>
|
||||||
|
<Extensions>qrc;*</Extensions>
|
||||||
|
<ParseFiles>false</ParseFiles>
|
||||||
|
</Filter>
|
||||||
|
<Filter Include="Generated Files">
|
||||||
|
<UniqueIdentifier>{71ED8ED8-ACB9-4CE9-BBE1-E00B30144E11}</UniqueIdentifier>
|
||||||
|
<Extensions>moc;h;cpp</Extensions>
|
||||||
|
<SourceControlFiles>False</SourceControlFiles>
|
||||||
|
</Filter>
|
||||||
|
</ItemGroup>
|
||||||
|
<ItemGroup>
|
||||||
|
<ClCompile Include="main.cpp">
|
||||||
|
<Filter>Source Files</Filter>
|
||||||
|
</ClCompile>
|
||||||
|
<ClCompile Include="AlgoTest.cpp">
|
||||||
|
<Filter>Source Files</Filter>
|
||||||
|
</ClCompile>
|
||||||
|
<ClCompile Include="QAlgDetect.cpp">
|
||||||
|
<Filter>Source Files</Filter>
|
||||||
|
</ClCompile>
|
||||||
|
<ClCompile Include="..\..\src\tpMain\algela\lpImgViewer.cpp">
|
||||||
|
<Filter>Source Files</Filter>
|
||||||
|
</ClCompile>
|
||||||
|
</ItemGroup>
|
||||||
|
<ItemGroup>
|
||||||
|
<QtMoc Include="AlgoTest.h">
|
||||||
|
<Filter>Header Files</Filter>
|
||||||
|
</QtMoc>
|
||||||
|
<QtMoc Include="QAlgDetect.h">
|
||||||
|
<Filter>Header Files</Filter>
|
||||||
|
</QtMoc>
|
||||||
|
<QtMoc Include="..\..\src\tpMain\algela\lpImgViewer.h">
|
||||||
|
<Filter>Header Files</Filter>
|
||||||
|
</QtMoc>
|
||||||
|
</ItemGroup>
|
||||||
|
<ItemGroup>
|
||||||
|
<QtUic Include="AlgoTest.ui">
|
||||||
|
<Filter>Form Files</Filter>
|
||||||
|
</QtUic>
|
||||||
|
</ItemGroup>
|
||||||
|
<ItemGroup>
|
||||||
|
<QtRcc Include="AlgoTest.qrc">
|
||||||
|
<Filter>Resource Files</Filter>
|
||||||
|
</QtRcc>
|
||||||
|
</ItemGroup>
|
||||||
|
</Project>
|
||||||
@ -0,0 +1,244 @@
|
|||||||
|
#include "QAlgDetect.h"
|
||||||
|
#include "ED.h"
|
||||||
|
#include "EDLines.h"
|
||||||
|
#include "EDCircles.h"
|
||||||
|
#include <QDebug>
|
||||||
|
#include <vector>
|
||||||
|
#include "CircleDetector.h"
|
||||||
|
|
||||||
|
using namespace std;
|
||||||
|
|
||||||
|
static QImage cvMat2QImage(const cv::Mat& mat) {
|
||||||
|
if (mat.type() == CV_8UC1) {
|
||||||
|
QImage image(mat.cols, mat.rows, QImage::Format_Indexed8);
|
||||||
|
image.setColorCount(256);
|
||||||
|
for (int i = 0; i < 256; i++) {
|
||||||
|
image.setColor(i, qRgb(i, i, i));
|
||||||
|
}
|
||||||
|
uchar *pSrc = mat.data;
|
||||||
|
for (int row = 0; row < mat.rows; row++) {
|
||||||
|
uchar *pDest = image.scanLine(row);
|
||||||
|
memcpy(pDest, pSrc, mat.cols);
|
||||||
|
pSrc += mat.step;
|
||||||
|
}
|
||||||
|
return image;
|
||||||
|
}
|
||||||
|
else if (mat.type() == CV_8UC3) {
|
||||||
|
const uchar *pSrc = (const uchar*)mat.data;
|
||||||
|
QImage image(pSrc, mat.cols, mat.rows, mat.step, QImage::Format_RGB888);
|
||||||
|
if (image.isNull())
|
||||||
|
return QImage();
|
||||||
|
return image.rgbSwapped();
|
||||||
|
}
|
||||||
|
else if (mat.type() == CV_8UC4) {
|
||||||
|
const uchar *pSrc = (const uchar*)mat.data;
|
||||||
|
QImage image(pSrc, mat.cols, mat.rows, mat.step, QImage::Format_ARGB32);
|
||||||
|
return image.copy();
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
qDebug() << "ERROR: Mat could not be converted to QImage.";
|
||||||
|
return QImage();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static cv::Mat QImageToMat(QImage image) {
|
||||||
|
cv::Mat mat;
|
||||||
|
QImage::Format f = image.format();
|
||||||
|
switch (image.format())
|
||||||
|
{
|
||||||
|
case QImage::Format_ARGB32:
|
||||||
|
case QImage::Format_RGB32:
|
||||||
|
case QImage::Format_ARGB32_Premultiplied:
|
||||||
|
mat = cv::Mat(image.height(), image.width(), CV_8UC4, (void*)image.constBits(), image.bytesPerLine());
|
||||||
|
break;
|
||||||
|
case QImage::Format_RGB888:
|
||||||
|
mat = cv::Mat(image.height(), image.width(), CV_8UC3, (void*)image.constBits(), image.bytesPerLine());
|
||||||
|
cv::cvtColor(mat, mat, CV_BGR2RGB);
|
||||||
|
break;
|
||||||
|
case QImage::Format_Indexed8:
|
||||||
|
case QImage::Format_Grayscale8:
|
||||||
|
mat = cv::Mat(image.height(), image.width(), CV_8UC1, (void*)image.constBits(), image.bytesPerLine());
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
return mat;
|
||||||
|
}
|
||||||
|
|
||||||
|
Mat _EnhanImg_sharpen(const Mat &img) {
|
||||||
|
int mGainValue = 3;
|
||||||
|
float amount = mGainValue * 0.5f;
|
||||||
|
Mat blurred;
|
||||||
|
|
||||||
|
int mSmoothValue = 3;
|
||||||
|
GaussianBlur(img, blurred, Size(), mSmoothValue);
|
||||||
|
Mat lowContrastMask = abs(img - blurred) < 5;
|
||||||
|
Mat orResult = img * (1 + amount) + blurred * (-amount);
|
||||||
|
img.copyTo(orResult, lowContrastMask);
|
||||||
|
Mat openMat = orResult.clone();
|
||||||
|
return openMat;
|
||||||
|
}
|
||||||
|
|
||||||
|
Mat findEdge2(const Mat &Src)
|
||||||
|
{
|
||||||
|
Mat ret = Src.clone();
|
||||||
|
Mat kernel = (Mat_<float>(5, 5) <<
|
||||||
|
-1, -1, -1, -1, -1,
|
||||||
|
-1, -1, -1, -1, -1,
|
||||||
|
-1, -1, 24, -1, -1,
|
||||||
|
-1, -1, -1, -1, -1,
|
||||||
|
-1, -1, -1, -1, -1);
|
||||||
|
filter2D(Src, ret, CV_8UC1, kernel);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
QAlgDetect::QAlgDetect(QObject *parent)
|
||||||
|
: QObject(parent)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
QAlgDetect::~QAlgDetect()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
#define REAIZE 4
|
||||||
|
|
||||||
|
void QAlgDetect::detect(QImage img, AlgCallBack func)
|
||||||
|
{
|
||||||
|
Mat srcImg = QImageToMat(img);
|
||||||
|
if(srcImg.channels()!=1)
|
||||||
|
cv::cvtColor(srcImg, srcImg, CV_RGB2GRAY);
|
||||||
|
Mat detectImg;
|
||||||
|
Mat src = srcImg;
|
||||||
|
|
||||||
|
cv::resize(src, detectImg, cv::Size(src.cols / REAIZE, src.rows / REAIZE));
|
||||||
|
int bBaseX = detectImg.cols;
|
||||||
|
int bBaseY = detectImg.rows;
|
||||||
|
// equalizeHist(detectImg, detectImg);//根据图像的实际情况使用 图像暗一点不需要，亮一点的需要
|
||||||
|
detectImg = _EnhanImg_sharpen(detectImg);
|
||||||
|
|
||||||
|
EDCircles edcircles(detectImg);
|
||||||
|
vector<mCircle> EDCircle = edcircles.getCircles();
|
||||||
|
double maxR = 0;
|
||||||
|
int nIndex = -1;
|
||||||
|
for (int i = 0; i < EDCircle.size(); i++)
|
||||||
|
{
|
||||||
|
int startX = EDCircle[i].center.x - EDCircle[i].r;
|
||||||
|
int startY = EDCircle[i].center.y - EDCircle[i].r;
|
||||||
|
if (startX < 0 || startY < 0)
|
||||||
|
continue;
|
||||||
|
if (EDCircle[i].center.x + EDCircle[i].r > bBaseX || EDCircle[i].center.y + EDCircle[i].r > bBaseY)
|
||||||
|
continue;
|
||||||
|
if (EDCircle[i].r > maxR)
|
||||||
|
{
|
||||||
|
maxR = EDCircle[i].r;
|
||||||
|
nIndex = i;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (nIndex != -1)
|
||||||
|
{
|
||||||
|
int startX = EDCircle[nIndex].center.x * REAIZE - EDCircle[nIndex].r * REAIZE;
|
||||||
|
int startY = EDCircle[nIndex].center.y * REAIZE - EDCircle[nIndex].r* REAIZE;
|
||||||
|
double radius = EDCircle[nIndex].r;
|
||||||
|
int hight = 2 * radius * REAIZE;
|
||||||
|
|
||||||
|
QPointF centerP;
|
||||||
|
centerP.setX(EDCircle[nIndex].center.x * REAIZE);
|
||||||
|
centerP.setY(EDCircle[nIndex].center.y * REAIZE);
|
||||||
|
double r = radius * REAIZE;
|
||||||
|
Mat aa = DetectCircle(srcImg, centerP, r);
|
||||||
|
QImage rltImg = cvMat2QImage(aa);
|
||||||
|
if (func) {
|
||||||
|
func(rltImg);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
if (startX > 0 && startY > 0 && hight > 0 \
|
||||||
|
&& startX < src.cols &&startY < src.rows \
|
||||||
|
&& hight < src.cols&&hight < src.rows \
|
||||||
|
&& (startX + hight) < src.cols && (startY + hight) < src.rows)
|
||||||
|
{
|
||||||
|
Mat cutMat = src(Rect(startX, startY, hight, hight));
|
||||||
|
if (cutMat.data != NULL)
|
||||||
|
{
|
||||||
|
// if (hight < 50)
|
||||||
|
// return Mat();
|
||||||
|
|
||||||
|
// pCircle->ptCenter = center;
|
||||||
|
// pCircle->fRadius = radius * REAIZE;
|
||||||
|
|
||||||
|
//2021-05-10 增加图像大小判断 对超过900像素的图像进行再一次压缩
|
||||||
|
if (cutMat.cols >= 900 || cutMat.rows >= 900)
|
||||||
|
{
|
||||||
|
Mat newCutImg;
|
||||||
|
cv::resize(cutMat, newCutImg, cv::Size(cutMat.cols / REAIZE, cutMat.rows / REAIZE));
|
||||||
|
// pCircle->fRadius = pCircle->fRadius / REAIZE;
|
||||||
|
// return newCutImg;
|
||||||
|
|
||||||
|
QImage rltImg = cvMat2QImage(newCutImg);
|
||||||
|
if (func) {
|
||||||
|
func(rltImg);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
QImage rltImg = cvMat2QImage(cutMat);
|
||||||
|
if (func) {
|
||||||
|
func(rltImg);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
if (func)
|
||||||
|
{
|
||||||
|
func(img);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
Mat QAlgDetect::DetectCircle(Mat img, QPointF center, double radius)
|
||||||
|
{
|
||||||
|
Mat detectImg;
|
||||||
|
//equalizeHist(img, detectImg);
|
||||||
|
detectImg = img;
|
||||||
|
CircleDetector cd;
|
||||||
|
cd.setAlgoType(CircleDetector::PeakCircle);
|
||||||
|
cd.setEdgeWidth(3);
|
||||||
|
cd.setPolarity(Polarity::White2Black);
|
||||||
|
cd.setFindBy(FindBy::Best);
|
||||||
|
double difRadiusMin = radius - 100;
|
||||||
|
double difRadiusMax = radius + 150;
|
||||||
|
if (difRadiusMin < 0)
|
||||||
|
{
|
||||||
|
difRadiusMin = 0;
|
||||||
|
difRadiusMax = abs(center.y()-(img.cols / 2))-50;
|
||||||
|
}
|
||||||
|
|
||||||
|
cd.setRadii(difRadiusMin, difRadiusMax);
|
||||||
|
cd.setACThres(3);
|
||||||
|
vector<float> allScores;
|
||||||
|
Vec3f bestCircle;
|
||||||
|
float bestScore = cd.detectBest(detectImg, Point2f(center.x(), center.y()), bestCircle, &allScores);
|
||||||
|
if (abs(bestScore) <= FLT_EPSILON || bestCircle == Vec3f::zeros()) {
|
||||||
|
return Mat();
|
||||||
|
}
|
||||||
|
|
||||||
|
QPointF cen(bestCircle[0], bestCircle[1]);
|
||||||
|
double r = bestCircle[2];
|
||||||
|
Rect rect;
|
||||||
|
rect.x = cen.x() - r;
|
||||||
|
rect.y = cen.y() - r;
|
||||||
|
rect.width = 2 * r;
|
||||||
|
rect.height = 2 * r;
|
||||||
|
Mat s = img(rect);
|
||||||
|
return s;
|
||||||
|
}
|
||||||
@ -0,0 +1,23 @@
|
|||||||
|
#ifndef _QALGDETECT_H_
|
||||||
|
#define _QALGDETECT_H_
|
||||||
|
|
||||||
|
#include <QObject>
|
||||||
|
#include <opencv.hpp>
|
||||||
|
#include <QImage>
|
||||||
|
|
||||||
|
using namespace cv;
|
||||||
|
typedef std::function<void(QImage)> AlgCallBack;
|
||||||
|
class QAlgDetect : public QObject
|
||||||
|
{
|
||||||
|
Q_OBJECT
|
||||||
|
|
||||||
|
public:
|
||||||
|
QAlgDetect(QObject *parent = nullptr);
|
||||||
|
~QAlgDetect();
|
||||||
|
|
||||||
|
void detect(QImage img, AlgCallBack func);
|
||||||
|
Mat DetectCircle(Mat img, QPointF center, double radius);
|
||||||
|
private:
|
||||||
|
|
||||||
|
};
|
||||||
|
#endif
|
||||||
@ -0,0 +1,10 @@
|
|||||||
|
#include "AlgoTest.h"
|
||||||
|
#include <QtWidgets/QApplication>
|
||||||
|
|
||||||
|
int main(int argc, char *argv[])
|
||||||
|
{
|
||||||
|
QApplication a(argc, argv);
|
||||||
|
AlgoTest w;
|
||||||
|
w.show();
|
||||||
|
return a.exec();
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue