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