@@ -146,28 +146,28 @@ void generateDetectedBoxes3D(
146
146
// construct boxes3d failed
147
147
std::cerr << " lidar_centerpoint_tvm: construct boxes3d failed" << std::endl;
148
148
}
149
- std::vector<Box3D> det_boxes3d_nonms (num_det_boxes3d);
149
+ std::vector<Box3D> det_boxes3d_no_nms (num_det_boxes3d);
150
150
std::copy_if (
151
- boxes3d.begin (), boxes3d.end (), det_boxes3d_nonms .begin (),
151
+ boxes3d.begin (), boxes3d.end (), det_boxes3d_no_nms .begin (),
152
152
is_score_greater (config.score_threshold_ ));
153
153
154
154
// sort by score
155
- std::sort (det_boxes3d_nonms .begin (), det_boxes3d_nonms .end (), score_greater ());
155
+ std::sort (det_boxes3d_no_nms .begin (), det_boxes3d_no_nms .end (), score_greater ());
156
156
157
157
// suppress by NMS
158
158
std::vector<bool > final_keep_mask (num_det_boxes3d);
159
159
const auto num_final_det_boxes3d =
160
- circleNMS (det_boxes3d_nonms , config.circle_nms_dist_threshold_ , final_keep_mask);
160
+ circleNMS (det_boxes3d_no_nms , config.circle_nms_dist_threshold_ , final_keep_mask);
161
161
162
162
det_boxes3d.resize (num_final_det_boxes3d);
163
163
std::size_t box_id = 0 ;
164
164
for (std::size_t idx = 0 ; idx < final_keep_mask.size (); idx++) {
165
165
if (final_keep_mask[idx]) {
166
- det_boxes3d[box_id] = det_boxes3d_nonms [idx];
166
+ det_boxes3d[box_id] = det_boxes3d_no_nms [idx];
167
167
box_id++;
168
168
}
169
169
}
170
- // std::copy_if(det_boxes3d_nonms .begin(), det_boxes3d_nonms .end(), final_keep_mask.begin(),
170
+ // std::copy_if(det_boxes3d_no_nms .begin(), det_boxes3d_no_nms .end(), final_keep_mask.begin(),
171
171
// det_boxes3d.begin(), is_kept());
172
172
}
173
173
0 commit comments