首次用opencv识别图形和标识图形

最新推荐文章于 2024-09-04 22:00:35 发布

iscada

最新推荐文章于 2024-09-04 22:00:35 发布

阅读量441

点赞数 4

分类专栏：人工智能文章标签： opencv

本文链接：https://blog.csdn.net/iscada/article/details/140456699

版权

人工智能专栏收录该内容

1 篇文章 0 订阅

订阅专栏

1，采用opencv4.10.0库

2，采用vs2022 c++开发

3，对图形进行识别和标识图形

string path = "./2.png";
Mat img = imread(path);
Mat imgGray, imgBlur, imgCanny, imgDil;

   // Preprocessing
   cvtColor(img, imgGray, COLOR_BGR2GRAY);   // 灰度图
   imshow("imggray",imgGray);
   GaussianBlur(imgGray, imgBlur, Size(3, 3), 3, 0);   // 高斯模糊处理
   imshow("imgBlur", imgBlur);
   Canny(imgBlur, imgCanny, 25, 75);   // Canny边缘检测算法
   imshow("imgCanny", imgCanny);
   Mat kernel = getStructuringElement(MORPH_RECT, Size(3, 3));
   dilate(imgCanny, imgDil, kernel);   // 膨胀图
   imshow("imgDil", imgDil);
   // 基于膨胀图，在原图上绘制轮廓边界、绘制边界包围盒以及形状描述
   getContours(imgDil, img);

imshow("Image", img);

/// Color Detection //
// 获取轮廓边界、绘制边界包围盒、形状描述
void getContours(Mat imgDil, Mat img) {

vector<vector<Point>> contours;
vector<Vec4i> hierarchy;

   // 从膨胀化的二值图像中检索轮廓
   findContours(imgDil, contours, hierarchy, RETR_EXTERNAL, CHAIN_APPROX_SIMPLE);
   //drawContours(img, contours, -1, Scalar(255, 0, 255), 2);

vector<vector<Point>> conPoly(contours.size()); // 逼近的多边形曲线，接近contours轮廓多边形
vector<Rect> boundRect(contours.size()); // contours轮廓多边形的边界包围盒

   // 遍历每一个轮廓多边形
   for (int i = 0; i < contours.size(); i++)
   {
       int area = contourArea(contours[i]);   // 计算轮廓的面积
       cout << area << endl;
       string objectType;

       if (area > 1000)   // 过滤那些面积特别小的轮廓，消除噪声
       {
           float peri = arcLength(contours[i], true);   // 计算轮廓周长(封闭的或者非封闭的)或曲线长度
           approxPolyDP(contours[i], conPoly[i], 0.02 * peri, true);   // 以指定精度逼近多边形曲线
           cout << conPoly[i].size() << endl;
           boundRect[i] = boundingRect(conPoly[i]);   // 计算顶点集合或灰度图像的非零像素的右上边界矩形，获取边界包围盒

           int objCor = (int)conPoly[i].size();   // 轮廓多边形的角落（顶点）个数
           // 根据objCor判断轮廓多边形的形状类型
           if (objCor == 3) {
               objectType = "Tri";   // 三角形
           }
           else if (objCor == 4) { // 四边形
               float aspRatio = (float)boundRect[i].width / (float)boundRect[i].height;   // 边界包围盒的宽高比：宽度/高度
               cout << aspRatio << endl;
               if (aspRatio > 0.95 && aspRatio < 1.05) {   // 边界包围盒宽高比大于0.95，或者小于1.05，则认为是正方形
                   objectType = "Square";   // 正方形
               }
               else {
                   objectType = "Rect";   // 矩形
               }
           }
           else if (objCor > 4) {
               objectType = "Circle";   // 圆形
           }

           drawContours(img, conPoly, i, Scalar(255, 0, 255),2);   // 绘制轮廓或填充轮廓，颜色为粉色
           rectangle(img, boundRect[i].tl(), boundRect[i].br(), Scalar(0, 255, 0), 2);   // 绘制边界包围盒，颜色为绿色
           putText(img, objectType, { boundRect[i].x,boundRect[i].y - 5 }, FONT_HERSHEY_PLAIN, 1, Scalar(0, 69, 255), 2);   // 在边界包围盒左上方往上5像素的位置，绘制其形状的描述文字
       }
   }
}