利用opencv检测出矩形

最新推荐文章于 2024-01-21 08:39:56 发布

qq_31376113

最新推荐文章于 2024-01-21 08:39:56 发布

阅读量2.7k

点赞数 1

分类专栏： opencv c++

本文链接：https://blog.csdn.net/qq_31376113/article/details/50897470

版权

c++ 同时被 2 个专栏收录

5 篇文章 0 订阅

订阅专栏

opencv

2 篇文章 0 订阅

订阅专栏

利用opencv和c++语言来检测出矩形轮廓并且返回矩形的顶点坐标

#pragma once
#include "TROOT.h"
#include "CBinToProfile.h"
#include "TH1.h"
#include "CBmpDiff.h"
#include "CCircleToPeak.h"
#include "CColorCut.h"
#include "CCubicBSpline.h"
#include "CCurveAngle.h"
#include "CEdgeHit.h"
#include "CEdgeHitChain.h"
#include "CEdgeToCircleChain.h"
#include "CEdgeToFloat.h"
#include "CFloatMatch.h"
#include "CFrame.h"
#include "CIntProfileToFloat.h"
#include "CLBP.h"
#include "CLine.h"
#include "CLinePara.h"
#include "CNormToPeak.h"
#include "CPicToBin.h"
#include "CPicToEdge.h"
#include "CPointProjection.h"
#include "CPointToCircle.h"
#include "CPointToNorm.h"
#include "CPointToPeak.h"
#include "CProfile.h"
#include "CRegionMatch.h"
#include "CSixSword.h"
#include "CTime.h"
#include "lib.h"
#include "vlib.h"
#include "paras.h"
#include "spline.h"
#include <opencv2/opencv.hpp>
#include <iostream>
#include <windows.h>

using namespace cv;
using namespace std;

void setFrame(VideoCapture cap, double w, double h)
{
    cap.set(CV_CAP_PROP_FRAME_WIDTH, w);
    cap.set(CV_CAP_PROP_FRAME_HEIGHT, h);
}

CvMemStorage* storage;//CvMemStorage 动态内存存储及操作函数
CvMemStorage* tempStorage;
void initStorage()//初始化内存空间
{
    storage = cvCreateMemStorage(0);//分配内存空间
    tempStorage = cvCreateMemStorage(0);
}
void releaseStorage()//释放内存空间
{
    cvReleaseMemStorage (&storage);
    cvReleaseMemStorage (&tempStorage);
}

CvSeq* tempContour;
CvSeqWriter  writer;
CEdgeHit *tempHit;
CvPoint tempPoint;

CvSeq* transform(CEdgeHitChain chain)//自定义的transform函数
{
    tempContour = cvCreateSeq(CV_SEQ_POLYGON, sizeof(CvSeq), sizeof(CvPoint), tempStorage);//创建序列
    cvStartAppendToSeq( tempContour, &writer );//序列写入操作把temppoint写入tempContour

    tempHit = chain.first;
    while (tempHit!=chain.last)
    {
        tempPoint.x = tempHit->x;
        tempPoint.y = tempHit->y;
        CV_WRITE_SEQ_ELEM( tempPoint, writer );//通过循环把tempPoint全部写入writer
        tempHit = tempHit->next;
    }
    return tempContour;//返回轮廓
}

// angle函数用来返回（两个向量之间找到角度的余弦值）勾股定理
double angle(CvPoint* pt1, CvPoint* pt2, CvPoint* pt0)
{
    double dx1 = pt1->x - pt0->x;
    double dy1 = pt1->y - pt0->y;
    double dx2 = pt2->x - pt0->x;
    double dy2 = pt2->y - pt0->y;
    return (dx1*dx2 + dy1*dy2) / sqrt((dx1*dx1 + dy1*dy1)*(dx2*dx2 + dy2*dy2) + 1e-10);
}

int main(int argc, char *argv[])
{
    int c;
    printf("welcome!\n");
    VideoCapture cap_1;
    cap_1.open(0);
    if (!cap_1.isOpened())
    {
        printf("摄像头1打开失败！\n");
        return -1;
    }
    double w = 320, h = 320;
    setFrame(cap_1, w, h);

    Mat frame_1;
    IplImage* image_1;
    IplImage* gray;
    CvSize sz;
    CSixSword tc;

    CvSeq* contour;
    CvSeq* result;
    CvSeq* squares;

    initStorage();
    bool stop = false;
    while (!stop)
    {
        cout << "----------------------------------" << endl;
        cap_1 >> frame_1;
        GaussianBlur(frame_1, frame_1, cv::Size(5, 5), 0, 0);//高斯滤波
        image_1 = &IplImage(frame_1);
        //image_1 = cvLoadImage("pic.png", 1);
        //image_1 = cvLoadImage("ggu.jpg", 1);
        //image_1 = cvLoadImage("ppo.png", 1);

        sz = cvSize(image_1->width, image_1->height);
        gray = cvCreateImage(sz, IPL_DEPTH_8U, 1);//创建头并分配数据
        cvCvtColor(image_1, gray, CV_BGR2GRAY);//灰度处理
        tc.setup(gray);//把灰度图传给tc
        tc.transform();
        cout << "v.size:" << tc.v.size() << endl;

        contour = cvCreateSeq(CV_SEQ_POLYGON, sizeof(CvSeq), sizeof(CvPoint), storage);//创建序列contour
        squares = cvCreateSeq(0, sizeof(CvSeq), sizeof(CvPoint), storage);//创建序列squares

        double s, t;
        for(int i = 0; i < tc.v.size(); i++)
        {
            contour = transform(tc.v[i]);//返回轮廓
            result = cvApproxPoly(contour, sizeof(CvContour), storage, CV_POLY_APPROX_DP, cvContourPerimeter(contour)*0.02, 0);//逼近函数
            if (result->total == 4 && cvCheckContourConvexity(result))//检测是否是四边形和凸边形
            {
                s = 0;
                for (int j = 2; j < 5; j++)//for循环来判断四边形的顶点是否都是直角
                {
                    t = fabs(angle(
                        (CvPoint*)cvGetSeqElem(result, j),
                        (CvPoint*)cvGetSeqElem(result, j - 2),
                        (CvPoint*)cvGetSeqElem(result, j - 1)));
                    s = s > t ? s : t;
                }
                if (s < 0.1)
                    for (int j = 0; j < 4; j++)
                        cvSeqPush(squares, (CvPoint*)cvGetSeqElem(result, j));//squares最后存储的就是矩形
            }
        }

        cout << "squares->total:" << squares->total/4 << endl << endl;

        CvSeqReader reader;
        cvStartReadSeq(squares, &reader, 0);
        int i;
        int x[4] = { 0 }, y[4] = { 0 };
        for (i = 0; i < squares->total; i += 4)//i+=4是因为四个点才是一矩形
        {
            CvPoint pt[4], *rect = pt;
            int count = 4;
            CV_READ_SEQ_ELEM(pt[0], reader);//for循环来读取顶点坐标
            CV_READ_SEQ_ELEM(pt[1], reader);
            CV_READ_SEQ_ELEM(pt[2], reader);
            CV_READ_SEQ_ELEM(pt[3], reader);
            cout << "--------square>>" << i/4+1 << endl;
            for (int i = 0; i<4; i++)
            {
                x[i] = rect[i].x;
                y[i] = rect[i].y;
                cout << "x" << i << ":" << x[i] << '\t';
                cout << "y" << i << ":" << y[i] << endl;
            }
            cvPolyLine(image_1, &rect, &count, 1, 1, CV_RGB(0, 255, 0), 1, CV_AA, 0);//在图像中用绿色线标出矩形区域
        }
        cvShowImage("Vedio_1", image_1);//显示窗口
        if (waitKey(27) >= 0)//Esc键退出
            stop = true;
    }
    releaseStorage();//释放内存
    return 0;
}