Skip to content

Commit 31d7f11

Browse files
committed
fix: highest confidence
Signed-off-by: tzhong518 <sworgun@gmail.com>
1 parent e4cae6f commit 31d7f11

File tree

1 file changed

+11
-7
lines changed
  • perception/traffic_light_multi_camera_fusion/src

1 file changed

+11
-7
lines changed

perception/traffic_light_multi_camera_fusion/src/node.cpp

+11-7
Original file line numberDiff line numberDiff line change
@@ -82,13 +82,17 @@ int compareRecord(const traffic_light::FusionRecord & r1, const traffic_light::F
8282
int visible_score_1 = calVisibleScore(r1);
8383
int visible_score_2 = calVisibleScore(r2);
8484
if (visible_score_1 == visible_score_2) {
85-
int area_1 = r1.roi.roi.width * r1.roi.roi.height;
86-
int area_2 = r2.roi.roi.width * r2.roi.roi.height;
87-
if (area_1 < area_2) {
88-
return -1;
89-
} else {
90-
return static_cast<int>(area_1 > area_2);
91-
}
85+
double sum_1 = 0;
86+
for_each(r1.signal.elements.begin(), r1.signal.elements.end(), [&sum_1](const auto & elem) {
87+
sum_1 = sum_1 + elem.confidence;
88+
});
89+
double sum_2 = 0;
90+
for_each(r2.signal.elements.begin(), r2.signal.elements.end(), [&sum_2](const auto & elem) {
91+
sum_2 = sum_2 + elem.confidence;
92+
});
93+
double confidence_1 = sum_1 / r1.signal.elements.size();
94+
double confidence_2 = sum_2 / r2.signal.elements.size();
95+
return confidence_1 < confidence_2 ? -1 : 1;
9296
} else {
9397
return visible_score_1 < visible_score_2 ? -1 : 1;
9498
}

0 commit comments

Comments
 (0)