简体   繁体   中英

display rectangle around components in original image

I am using OpenCV for a C++ application.I'm using connected component for object detection.I want to draw a rectangle around object in original frame.I can draw rectangle in comonent window.can I draw a color rectangle in gray scale image ?in below I write part of my code.thanks for your help.

Mat frame;
Mat stat, centroid;
int threshval = 100;
static void on_trackbar(int, void*){
 Mat bw = threshval < 128 ? (frame < threshval) : (frame > threshval);
 Mat labelImage(frame.size(), CV_32S);
 int nLabels = connectedComponentsWithStats(bw, labelImage, stat, centroid, 8);
 std::vector<Vec3b> colors(nLabels);
      colors[0] = Vec3b(0, 0, 0);//background
   for (int label = 1; label < nLabels; ++label) {
 colors[label] = Vec3b((rand() & 255), (rand() & 255), (rand() & 255));}
 at dst(frame.size(), CV_8UC3);
 for (int r = 0; r < dst.rows; ++r) {
     for (int c = 0; c < dst.cols; ++c) {
         int label = labelImage.at<int>(r, c);
         Vec3b &pixel = dst.at<Vec3b>(r, c);

         pixel = colors[label];} 
for (int i = 0;i < nLabels;i++)
    {

        vector<Rect> rComp;
        rComp.push_back(Rect(Point((stat.at<int>(i, CC_STAT_LEFT) ), (stat.at<int>(i, CC_STAT_TOP) )), Size((stat.at<int>(i, CC_STAT_WIDTH) ), (stat.at<int>(i, CC_STAT_HEIGHT)))));
    //  

            rectangle(dst, Rect(Point(stat.at<int>(i, CC_STAT_LEFT  ) , stat.at<int>(i, CC_STAT_TOP  ) ), Size(stat.at<int>(i, CC_STAT_WIDTH   ) , stat.at<int>(i, CC_STAT_HEIGHT  ))), Scalar(0, 255, 255));}
}

    for (int i = 0;i < nLabels;i++) {
        int x = stat.at<int>(i, CC_STAT_LEFT);
        int y = stat.at<int>(i, CC_STAT_TOP);
        int w = stat.at<int>(i, CC_STAT_WIDTH) ;
        int h = stat.at<int>(i, CC_STAT_HEIGHT);
        rectangle(frame, Rect(x,y,w,h), Scalar(0, 255, 255));
    }
}
imshow("Connected Components", dst);

As mentioned here and here , you can't draw colored rectangles on gray scale images. You could either use Scalar(255, 255, 255) - white / Scalar(0, 0, 0) or follow the hack in the first link.

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM