Skip to content

Warn and ignore invalid predictions from mask_rcnn adapter #3998

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
pwolnows merged 3 commits into from
Feb 20, 2025
Merged

Warn and ignore invalid predictions from mask_rcnn adapter #3998

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
6ae1c22
Skip invalid mask_rcnn predictions and display warning
pwolnows Feb 20, 2025
dd29d51
Update formating mask_rcnn.py
pwolnows Feb 20, 2025
236a334
Update typo in mask_rcnn.py
pwolnows Feb 20, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 15 additions & 1 deletion tools/accuracy_checker/accuracy_checker/adapters/mask_rcnn.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -167,6 +167,20 @@ def _process_tf_obj_detection_api_outputs(self, raw_outputs, identifiers, frame_
im_scores = im_scores[:num_valid_detections]
im_boxes = im_boxes[:num_valid_detections]
im_raw_masks = im_raw_masks[:num_valid_detections]
invalid_scores = np.logical_or(im_scores > 1, im_scores <= 0)
if np.sum(invalid_scores) > 0:
warnings.warn(
f"Output for {identifier} contains invalid scores and boxes!\n"
f"Scores: {im_scores[ivalid_boxes]}\n"
f"Boxes: {im_boxes[ivalid_boxes]}!"
)
im_classes = im_classes[~invalid_scores]
im_scores = im_scores[~invalid_scores]
im_boxes = im_boxes[~invalid_scores]
im_raw_masks = im_raw_masks[~invalid_scores]
if len(im_classes) == 0:
continue

original_image_size = image_meta['image_size'][:2]
im_boxes[:, 1::2] *= original_image_size[1]
im_boxes[:, 0::2] *= original_image_size[0]
Expand All @@ -190,7 +204,7 @@ def _process_tf_obj_detection_api_outputs(self, raw_outputs, identifiers, frame_
'segmentation_prediction': instance_segmentation_prediction
}))

return results
return results

def _process_pytorch_outputs(self, raw_outputs, identifiers, frame_meta):
if self.boxes_out not in raw_outputs:
Expand Down
Loading