OPENCV入门之启动摄像头《分水岭算法》

来源:互联网 发布:mac srt 编辑:程序博客网 时间:2024/05/16 04:35

首先这代码是改编过来,原本是不能自启的,下面直接上代码吧,基础的东西还是要机子动手多打代码,多理解,虽然我是刚入门的,如果有错误或者有更好的想法大家一块交流,谢谢。

 
#include "opencv2/imgproc/imgproc.hpp"  
#include "opencv2/highgui/highgui.hpp"  


#include <iostream>  
  
using namespace cv;  
using namespace std;  


  
Vec3b RandomColor(int value);  //生成随机颜色函数  
  
int main( int argc, char* argv[] )  
{
system("color 1E");
Mat frame;
        Mat edges;
    VideoCapture cap(0);  
    if(!cap.isOpened())  
    {  
        return -1;  
    }  
 Mat srcImage,grayImage;
    bool stop = false;  
    while(!stop)  
    {  
        cap>>frame;  
cvtColor(frame,edges,CV_RGB2GRAY);
GaussianBlur(edges,edges,Size(5,5),2);
imshow("GaussianBlur",edges);
Canny(edges,edges,80,150);
imshow("Canny",edges);
vector<vector<Point>>contours;
vector<Vec4i>hierarchy;
findContours(edges,contours,hierarchy,RETR_TREE,CHAIN_APPROX_SIMPLE,Point());
Mat imageContours=Mat::zeros(frame.size(),CV_8UC1);
Mat marks(frame.size(),CV_32S);
marks=Scalar::all(0);
int index=0;
int compCount = 0;
for( ; index >= 0; index = hierarchy[index][0], compCount++ )   
    {  
        //对marks进行标记,对不同区域的轮廓进行编号,相当于设置注水点,有多少轮廓,就有多少注水点  
        drawContours(marks, contours, index, Scalar::all(compCount+1), 1, 8, hierarchy);  
        drawContours(imageContours,contours,index,Scalar(255),1,8,hierarchy);    
    }
        Mat marksShows;
convertScaleAbs(marks,marksShows);
imshow("marksShows",marksShows);
imshow("轮廓",imageContours);
watershed(frame,marks);

Mat afterWatershed;
convertScaleAbs(marks,afterWatershed);
imshow("After",afterWatershed);
Mat PerspectiveImage=Mat::zeros(frame.size(),CV_8UC3);
for(int i=0;i<marks.rows;i++)
{
for(int j=0;j<marks.rows;j++)
{
int index=marks.at<int>(i,j);
if(marks.at<int>(i,j)==-1)
{
PerspectiveImage.at<Vec3b>(i,j)=Vec3b(255,255,255);
}
else
{
PerspectiveImage.at<Vec3b>(i,j)=RandomColor(index);
}
}
}
     imshow("colorfill", PerspectiveImage);
Mat wshed;
addWeighted(frame,0.4, PerspectiveImage,0.6,0,wshed);
imshow("frame",wshed);
waitKey();
}
 }
Vec3b RandomColor(int value)
{
value=value%255;
RNG rng;
int aa=rng.uniform(0,value);
int bb=rng.uniform(0,value);
int cc=rng.uniform(0,value);
return Vec3b(aa,bb,cc);
}

原创粉丝点击