cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: check if roi width and height is NaN

Browse Source
This commit is contained in:
homuler 2021-08-26 12:27:46 +09:00
parent 710fb3de58
commit f1034bd5a8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mediapipe/calculators/tensor/tensors_to_detections_calculator.cc
View File

@ -670,7 +670,7 @@ absl::Status TensorsToDetectionsCalculator::ConvertToDetections(
detection_boxes[box_offset + 2], detection_boxes[box_offset + 3], detection_boxes[box_offset + 2], detection_boxes[box_offset + 3],
detection_scores[i], detection_classes[i], options_.flip_vertically()); detection_scores[i], detection_classes[i], options_.flip_vertically());
const auto& bbox = detection.location_data().relative_bounding_box(); const auto& bbox = detection.location_data().relative_bounding_box();
if (bbox.width() < 0 || bbox.height() < 0) { if (bbox.width() < 0 || bbox.height() < 0 || std::isnan(bbox.width()) || std::isnan(bbox.height())) {
// Decoded detection boxes could have negative values for width/height due // Decoded detection boxes could have negative values for width/height due
// to model prediction. Filter out those boxes since some downstream // to model prediction. Filter out those boxes since some downstream
// calculators may assume non-negative values. (b/171391719) // calculators may assume non-negative values. (b/171391719)