自动透视校正为四边形对象

参考opencv开源上的，跑通了注释了一下，以便后续学习

#include <opencv2/imgproc/imgproc.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <iostream>

cv::Point2f center(0,0);
cv::Point2f computeIntersect(cv::Vec4i a,cv::Vec4i b)
{
int x1 = a[0],y1 = a[1],x2 = a[2],y2 = a[3],x3 = b[0],y3 = b[1],x4 = b[2],y4 = b[3];
if (float d = ((float)(x1 - x2)*(y3 - y4)-(y1 - y2)*(x3 - x4)))
{
cv::Point2f pt;
pt.x = ((x1*y2 - y1*x2)*(x3 - x4) - (x1 - x2)*(x3*y4 - y3*x4))/d;
pt.y = ((x1*y2 - y1*x2)*(y3 - y4) - (y1 - y2)*(x3*y4 - y3*x4))/d;
return pt;
}
else
return cv::Point2f(-1,-1);
}
void sortCorners(std::vector<cv::Point2f>& corners,cv::Point2f center)
{
std::vector<cv::Point2f> top,bot;
for (unsigned int i =0;i< corners.size();i++)
{
if (corners[i].y<center.y)
{
top.push_back(corners[i]);
}
else
{
bot.push_back(corners[i]);
}
}
cv::Point2f tl = top[0].x > top[1].x ? top[1] : top[0];
cv::Point2f tr = top[0].x > top[1].x ? top[0] : top[1];
cv::Point2f bl = bot[0].x > bot[1].x ? bot[1] : bot[0];
cv::Point2f br = bot[0].x > bot[1].x ? bot[0] : bot[1];
corners.clear();
//注意以下存放顺序是顺时针，当时这里出错了，如果想任意顺序下文开辟的四边形矩阵注意对应
corners.push_back(tl);
corners.push_back(tr);
corners.push_back(br);
corners.push_back(bl);
}

算法流程：先彩色转灰度，然后模糊求canny边缘，再用hough检测直线，求出四线交点，利用opencv自带的求透视矩阵的函数求出透视矩阵，然后利用透视矩阵转换源图像所需的四边形，效果图后续见图片：

int main()
{
cv::Mat src = cv::imread("C:/Users/Public/Pictures/Sample Pictures/perspective-quadrilateral-src-img.jpg");
if (src.empty())
{
return -1;
}
cv::Mat bw;
cv::cvtColor(src,bw,CV_BGR2GRAY);
imshow("gray_src",bw);
cv::blur(bw,bw,cv::Size(3,3));
imshow("blur",bw);
cv::Canny(bw,bw,100,100,3);
imshow("cannyblur",bw);
std::vector<cv::Vec4i> lines;
cv::HoughLinesP(bw,lines,1,CV_PI/180,70,30,10);
//1像素分辨能力  1度的角度分辨能力        >70可以检测成连线       30是最小线长
//在直线L上的点（且点与点之间距离小于maxLineGap=10的）连成线段，然后这些点全部删除，并且记录该线段的参数，就是起始点和终止点
//needed for visualization only//这里是将检测的线调整到延长至全屏，即射线的效果，其实可以不必这么做
for (unsigned int i = 0;i<lines.size();i++)
{
cv::Vec4i v = lines[i];
lines[i][0] = 0;
lines[i][1] = ((float)v[1] - v[3])/(v[0] - v[2])* -v[0] + v[1];
lines[i][2] = src.cols;
lines[i][3] = ((float)v[1] - v[3])/(v[0] - v[2])*(src.cols - v[2]) + v[3];
}
std::vector<cv::Point2f> corners;//线的交点存储
for (unsigned int i = 0;i<lines.size();i++)
{
for (unsigned int j=i+1;j<lines.size();j++)
{
cv::Point2f pt = computeIntersect(lines[i],lines[j]);
if (pt.x >= 0 && pt.y >=0)
{
corners.push_back(pt);
}
}
}
std::vector<cv::Point2f> approx;
cv::approxPolyDP(cv::Mat(corners),approx,cv::arcLength(cv::Mat(corners),true)*0.02,true);
if (approx.size()!=4)
{
std::cout<<"The object is not quadrilateral（四边形）!"<<std::endl;
return -1;
}
//get mass center
for (unsigned int i = 0;i < corners.size();i++)
{
center += corners[i];
}
center *=(1./corners.size());
sortCorners(corners,center);
cv::Mat dst = src.clone();
//Draw Lines
for (unsigned int i = 0;i<lines.size();i++)
{
cv::Vec4i v = lines[i];
cv::line(dst,cv::Point(v[0],v[1]),cv::Point(v[2],v[3]),CV_RGB(0,255,0));    //目标版块画绿线
}
//draw corner points
cv::circle(dst,corners[0],3,CV_RGB(255,0,0),2);
cv::circle(dst,corners[1],3,CV_RGB(0,255,0),2);
cv::circle(dst,corners[2],3,CV_RGB(0,0,255),2);
cv::circle(dst,corners[3],3,CV_RGB(255,255,255),2);
//draw mass center
cv::circle(dst,center,3,CV_RGB(255,255,0),2);
cv::Mat quad = cv::Mat::zeros(300,220,CV_8UC3);//设定校正过的图片从320*240变为300*220
//corners of the destination image
std::vector<cv::Point2f> quad_pts;
quad_pts.push_back(cv::Point2f(0,0));
quad_pts.push_back(cv::Point2f(quad.cols,0));//(220,0)
quad_pts.push_back(cv::Point2f(quad.cols,quad.rows));//(220,300)
quad_pts.push_back(cv::Point2f(0,quad.rows));
// Get transformation matrix
cv::Mat transmtx = cv::getPerspectiveTransform(corners,quad_pts);  //求源坐标系（已畸变的）与目标坐标系的转换矩阵
// Apply perspective transformation透视转换
cv::warpPerspective(src,quad,transmtx,quad.size());
cv::imshow("src",src);
cv::imshow("image",dst);
cv::imshow("quadrilateral",quad);
cv::waitKey();
return 0;
}

参考网址http://opencv-code.com/tutorials/automatic-perspective-correction-for-quadrilateral-objects/

自动透视校正为四边形对象相关推荐

Autolisp：利用AuoCAD之Lisp编程案例之自动智能获取所选对象的面积并标注在指定位置
Autolisp:利用AuoCAD之Lisp编程案例之自动智能获取所选对象的面积并标注在指定位置目录输出结果实现代码输出结果实现代码 ;; write area value of a clo ...
OpenCV透视校正perspective correction的实例(附完整代码)
OpenCV透视校正perspective correction的实例 OpenCV透视校正perspective correction的实例 OpenCV透视校正perspective correc ...
图形学基础之透视校正插值
透视校正插值 (Perspective-Correct Interpolation) 问题的提出在使用光栅化的图形学方法中,法线,颜色,纹理坐标这些属性通常是绑定在图元的顶点上的.在3D空间中,这些 ...
自动将Map转换成对象的方法
实现自动将Map转换成对象,借助第三方工具commons-beanutils-1.8.0.jar+commons-logging.jar实现,需要遵守的规则创建的对象的属性名必须与map的key相同. ...
python实现图像自动Gamma校正
python实现图像自动Gamma校正前言: 关于Gamma: Gamma校正是对动态范围内亮度的非线性存储/还原算法,即输入值进行的非线性操作,使输出值与输入值呈指数关系:从效果上来说Gamma校 ...
自动控制原理学习笔记---自动控制系统校正与综合
自控原理学习笔记自控原理学习笔记专栏第一章--反馈控制系统的动态模型第二章--控制系统稳定性分析第三章--连续时间系统性能分析第四章--自动控制系统校正与综合第五章--线性离散系统自动控 ...
【图形学】关于透视校正
关于透视校正使用如下的矩阵进行透视投影: p r o j = [ n 0 0 0 0 n 0 0 0 0 n + f n f 0 0 − 1 0 ] \mathrm{proj}=\begin{bma ...
图片透视校正 java js_iOS 使用OpenCV 实现图片的透视校正
孤独的香蕉.png 一.前言写这篇博客的目的为记录,并且供大家学习,希望大家多多交流,有不正确的地方请指正,轻喷.!~~ 二.透视校正简介透视校正或者称为透视控制是对照片进行合成或者编辑以得到符 ...
java 日志生成器_自动生成 java 测试 mock 对象框架 DataFactory-01-入门使用教程
项目简介 Data-Factory 用于根据对象,随机自动生成初始化信息,避免了手动创建对象的繁琐,便于测试. 特性 8 大基本类型的支持 String.Date.金额,日期等常见类型的支持 java ...

自动透视校正为四边形对象

自动透视校正为四边形对象相关推荐

最新文章

热门文章