OpenCV学习代码记录——轮廓(contour)检测
很久之前学习过一段时间的OpenCV,当时没有做什么笔记,但是代码都还在,这里把它贴出来做个记录。
代码放在码云上,地址在这里https://gitee.com/solym/OpenCVTest/tree/master/OpenCVTest
效果
代码
#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc/imgproc.hpp> // cvtColor
int contours_test();
int ContourDetection()
{
//return contours_test();
cv::Mat image; // 加载原始图片
cv::Mat gary; // 存储灰度图像
cv::Mat dstimg; // 绘制轮廓目标图片
// 创建两个窗口
cv::namedWindow("src"); // 原始图片显示窗口
cv::namedWindow("dst"); // 轮廓图片显示窗口
// 载入原始图片
image = cv::imread("../Image/sisy.jpg");
if (image.empty()) {
puts("图片加载失败");
return -1;
}
cv::imshow("src", image); // 显示原始图片
gary.create(image.size(), CV_8U); // 申请灰度图存储空间
cv::cvtColor(image, gary, CV_BGR2GRAY); // 转换原始图为灰度图
cv::threshold(gary, gary, 128, 255, cv::THRESH_BINARY); // 转换为二值图
std::vector<std::vector<cv::Point> > contours; // 检测的轮廓数组
std::vector<cv::Vec4i> hierarchy; //
int mode = CV_RETR_EXTERNAL; // 轮廓检测模式
//mode表示轮廓的检索模式
// CV_RETR_EXTERNAL表示只检测外轮廓
// CV_RETR_LIST检测的轮廓不建立等级关系
// CV_RETR_CCOMP建立两个等级的轮廓,上面的一层为外边界,里面的一层为内孔的边界信息。如果内孔内还有一个连通物体,这个物体的边界也在顶层。
// CV_RETR_TREE建立一个等级树结构的轮廓。具体参考contours.c这个demo
int method = CV_CHAIN_APPROX_SIMPLE;
//method为轮廓的近似办法
// CV_CHAIN_APPROX_NONE存储所有的轮廓点,相邻的两个点的像素位置差不超过1,即max(abs(x1 - x2),abs(y2 - y1)) == 1
// CV_CHAIN_APPROX_SIMPLE压缩水平方向,垂直方向,对角线方向的元素,只保留该方向的终点坐标,例如一个矩形轮廓只需4个点来保存轮廓信息
// CV_CHAIN_APPROX_TC89_L1,CV_CHAIN_APPROX_TC89_KCOS使用teh-Chinl chain 近似算法
// 查找contour
cv::findContours(gary/*输入图像(必须为一个2值单通道图像)*/,
contours/*, hierarchy*/, mode, method);
// 为轮廓显示图片申请空间
dstimg = cv::Mat(image.size(), CV_8UC3); // 3通道图像,以便彩色显示
image.copyTo(dstimg); // 拷贝源图像
// 将轮廓画出
cv::drawContours(dstimg/*目标图像*/,
contours/*输入的轮廓组*/,
-1 /*指明画第几个轮廓(负值表示全部轮廓)*/,
cv::Scalar(0,0,255)/*轮廓颜色BGR(此处以红色绘制)*/,
2 /*轮廓线宽*/,
8 /*轮廓线型*/,
cv::noArray()/*轮廓结构信息*/);
// 显示轮廓图片
cv::imshow("dst", dstimg);
// 等待按键
cv::waitKey();
}
int contours_test()
{
std::string image_name = "../Image/sisy.jpg";
cv::Mat src = cv::imread(image_name);
cv::imshow("src", src);
cv::Mat gray(src.size(), CV_8U);
cv::cvtColor(src, gray, CV_BGR2GRAY);//转换成灰度图
cv::imshow("gray", gray);
cv::threshold(gray, gray, 128, 255, cv::THRESH_BINARY);//转换成2值图像
cv::imshow("binary", gray);
/////////////////////////////////////////////////////////////////////
std::vector<std::vector<cv::Point>> contours;
cv::findContours(gray,
contours, // a vector of contours
CV_RETR_EXTERNAL, // retrieve the external contours
CV_CHAIN_APPROX_NONE); // retrieve all pixels of each contours
// Print contours' length
// std::cout << "Contours: " << contours.size() << std::endl;
std::vector<std::vector<cv::Point>>::const_iterator itContours = contours.begin();
for (; itContours != contours.end(); ++itContours)
{
//std::cout << "Size: " << itContours->size() << std::endl;
}
// draw black contours on white image
cv::Mat result(gray.size(), CV_8U, cv::Scalar(255));
cv::drawContours(result, contours,
-1, // draw all contours
cv::Scalar(0), // in black
2); // with a thickness of 2
cv::namedWindow("Contours");
cv::imshow("Contours", result);
// draw contours on the original image
cv::Mat original = cv::imread(image_name);
cv::drawContours(original, contours,
-1, // draw all contours
cv::Scalar(255, 255, 255), // in white
-1); // with a thickness of 2
cv::namedWindow("Contours on Animals");
cv::imshow("Contours on Animals", original);
// Let's now draw black contours on white image
result.setTo(cv::Scalar(255));
cv::drawContours(result, contours,
-1, // draw all contours
cv::Scalar(0), // in black
-1); // with a thickness of 1
//image= cv::imread("test.png",0);
cv::waitKey(0);
return 0;
}