commit 61f21e853d80a47e4d887f796905aeeb220d19f7
author milind-u <milind.upadhyay@gmail.com> Sun Jan 23 18:34:11 2022 -0800
committer Milind Upadhyay <milind.upadhyay@gmail.com> Sat Jan 29 13:40:47 2022 -0800
tree c65be505dd85f07e8563a5bc9c9d8c7fa75bdc3d
parent aaa4bc6f8d2f9a7329e65c77f80b9a1f500cb12b

Filter blobs based on aspect ratio and size

Signed-off-by: milind-u <milind.upadhyay@gmail.com>
Change-Id: Idad861a777a9f09de7da10792ce5ab73c5daa6d1

y2022/vision/viewer.cc

diff --git a/y2022/vision/viewer.cc b/y2022/vision/viewer.cc
index 8959f53..6573308 100644
--- a/y2022/vision/viewer.cc
+++ b/y2022/vision/viewer.cc
@@ -15,9 +15,7 @@
               "If set, capture a single image and save it to this filename.");
 DEFINE_string(channel, "/camera", "Channel name for the image.");
 DEFINE_string(config, "config.json", "Path to the config file to use.");
-DEFINE_string(png_dir,
-              "/home/jim/code/FRC/971-Robot-Code/y2020/vision/LED_Ring_exp",
-              "Path to a set of images to display.");
+DEFINE_string(png_dir, "LED_Ring_exp", "Path to a set of images to display.");
 DEFINE_bool(show_features, true, "Show the blobs.");
 
 namespace y2022 {
@@ -52,8 +50,8 @@
   }
 
   cv::Mat binarized_image, ret_image;
-  std::vector<std::vector<cv::Point>> unfiltered_blobs, filtered_blobs,
-      blob_stats;
+  std::vector<std::vector<cv::Point>> unfiltered_blobs, filtered_blobs;
+  std::vector<BlobDetector::BlobStats> blob_stats;
   BlobDetector::ExtractBlobs(rgb_image, binarized_image, ret_image,
                              filtered_blobs, unfiltered_blobs, blob_stats);
 
@@ -112,8 +110,8 @@
   for (auto file : file_list) {
     LOG(INFO) << "Reading file " << file;
     cv::Mat rgb_image = cv::imread(file.c_str());
-    std::vector<std::vector<cv::Point>> filtered_blobs, unfiltered_blobs,
-        blob_stats;
+    std::vector<std::vector<cv::Point>> filtered_blobs, unfiltered_blobs;
+    std::vector<BlobDetector::BlobStats> blob_stats;
     cv::Mat binarized_image =
         cv::Mat::zeros(cv::Size(rgb_image.cols, rgb_image.rows), CV_8UC1);
     cv::Mat ret_image =
@@ -127,7 +125,7 @@
     cv::imshow("image", rgb_image);
     cv::imshow("blobs", ret_image);
 
-    int keystroke = cv::waitKey(1000);
+    int keystroke = cv::waitKey(0);
     if ((keystroke & 0xFF) == static_cast<int>('q')) {
       return;
     }