前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >不废话,看我20行代码搞定色块提取与定位…….

不废话,看我20行代码搞定色块提取与定位…….

作者头像
OpenCV学堂
发布2020-03-09 13:36:29
1.2K0
发布2020-03-09 13:36:29
举报
文章被收录于专栏:贾志刚-OpenCV学堂

问题来由

这个问题是谁问我的我已经不记得了,刚开始的时候他发了这张图像给我,让我给他看一下,我当时告诉他转换一下色彩空间提取就好啦,后来我记得他在微信上有问了我一次,今天我整理文件看到这张图又想起了,感觉他问了我好几次我都没回复挺不意思的,但是我实在不知道他是谁了,微信上消息太多,早已经把他淹没了,加之我记忆力退化严重,思来想去只好写篇文章告诉他,我回答了!如果看完感觉对你也有用,点在看支持即可!

先看看他发我的图像文件吧

在来说说他的需求:

找到途中全部青色的区域,检测出来,绘制中心点!他用霍夫变换,结果直接翻车了,原因其实我很理解,这个图有很多梯度干扰,噪声干扰,用了肯定翻车!然后他就问我怎么办?

解题思路

直接转换到HSV色彩空间,得到如下结果:

然后我选择5x5的开操作,完成之后得到

使用OpenCV轮廓发现,对轮廓拟合圆,求的圆心坐标得到输出结果如下:

怎么样,效果好吗?

代码实现

上述步骤的代码演示,主要分为如下步骤

1.加载图像并转换到HSV色彩空间,得到mask 2.根据mask二值图像,进行形体学处理 3.使用轮廓发现,找到所有最外层轮廓 4.对轮廓进行圆拟合,得到圆心与半径,然后绘制

代码如下:

代码语言:javascript
复制
 1#include <opencv2/opencv.hpp>
 2#include <iostream>
 3
 4using namespace cv;
 5using namespace std;
 6
 7int main(int argc, char** argv) {
 8    Mat image = imread("D:/images/zsxq/zsxq_20.png");
 9    imshow("input", image);
10
11    // 色彩空间转换
12    Mat hsv, mask;
13    cvtColor(image, hsv, COLOR_BGR2HSV);
14    inRange(hsv, Scalar(40,43, 46), Scalar(50, 255, 255), mask);
15    imshow("mask", mask);
16    imwrite("D:/mask.png", mask);
17
18    // 形态学开操作
19    Mat se = getStructuringElement(MORPH_RECT, Size(5, 5), Point(-1, -1));
20    morphologyEx(mask, mask, MORPH_OPEN, se);
21    imshow("binary", mask);
22    imwrite("D:/binary.png", mask);
23
24    // 轮廓发现
25    vector<vector<Point>> contours;
26    vector<Vec4i> hiearchy;
27    findContours(mask, contours, hiearchy, RETR_EXTERNAL, CHAIN_APPROX_SIMPLE);
28    for (int i = 0; i < contours.size(); i++) {
29        // 圆拟合
30        RotatedRect rrt = fitEllipse(contours[i]);
31        Point ct = rrt.center;
32        int h = rrt.size.height;
33        int w = rrt.size.width;
34        printf("height : %d, width : %d \n", h, w);
35        circle(image, ct, 2, Scalar(0, 0, 255), 2, 8);
36        circle(image, ct, (h + w) / 4, Scalar(255, 0, 0), 2, 8, 0);
37    }
38
39    // 显示输出
40    imshow("result", image);
41    imwrite("D:/result.png", image);
42    waitKey(0);
43    return 0;
44}
45
本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2020-03-05,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 OpenCV学堂 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档