Merge pull request #151 from anarkiwi/minmax

add min max of predicted bbox region.
IQTLabs · Nov 29, 2023 · adfb7bf · adfb7bf
2 parents cd21811 + bea6765
commit adfb7bf
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/lib/image_inference_impl.cc b/lib/image_inference_impl.cc
@@ -476,10 +476,14 @@ void image_inference_impl::get_inference_() {
                 cv::Rect rssi_rect(int(tlx * xf), int(tly * yf), int(w * xf),
                                    int(h * yf));
                 cv::Mat rssi_points = (*output_item.points_buffer)(rssi_rect);
+                double rssi_min, rssi_max;
+                cv::minMaxLoc(rssi_points, &rssi_min, &rssi_max);
                 float rssi = cv::mean(rssi_points)[0];
                 auto &augmented = results_json[prediction_class.key()][i];
                 augmented["rssi"] = rssi;
                 augmented["rssi_samples"] = rssi_points.cols * rssi_points.rows;
+                augmented["rssi_min"] = rssi_min;
+                augmented["rssi_max"] = rssi_max;
                 if (rssi >= min_peak_points_) {
                   ++rendered_predictions;
                   cv::rectangle(*output_item.image_buffer, bbox_rect, white);

diff --git a/python/iqtlabs/qa_image_inference.py b/python/iqtlabs/qa_image_inference.py
@@ -331,7 +331,7 @@ def test_instance(self):
                 self.assertGreaterEqual(rssi_max, rssi_mean, metadata_result)
                 self.assertTrue(os.path.exists(metadata_result["image_path"]))
                 self.assertTrue(os.path.exists(metadata_result["predictions_image_path"]))
-                for k in ("rssi", "rssi_samples"):
+                for k in ("rssi", "rssi_samples", "rssi_min", "rssi_max"):
                     del result["predictions"]["modulation"][0][k]
                 self.assertEqual(result["predictions"], predictions_result)