You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

43 lines
1.3 KiB
Matlab

%helperSanitizeBoxes Sanitize box data.
% This example helper is used to clean up invalid bounding box data. Boxes
% with values <= 0 are removed and fractional values are rounded to
% integers.
%
% If none of the boxes are valid, this function passes the data through to
% enable downstream processing to issue proper errors.
% Copyright 2020 The Mathworks, Inc.
function boxes = helperSanitizeBoxes(boxes, imageSize)
persistent hasInvalidBoxes
valid = all(boxes > 0, 2);
if any(valid)
if ~all(valid) && isempty(hasInvalidBoxes)
% Issue one-time warning about removing invalid boxes.
hasInvalidBoxes = true;
warning('Removing ground truth bouding box data with values <= 0.')
end
boxes = boxes(valid,:);
boxes = roundFractionalBoxes(boxes, imageSize);
end
end
function boxes = roundFractionalBoxes(boxes, imageSize)
% If fractional data is present, issue one-time warning and round data and
% clip to image size.
persistent hasIssuedWarning
allPixelCoordinates = isequal(floor(boxes), boxes);
if ~allPixelCoordinates
if isempty(hasIssuedWarning)
hasIssuedWarning = true;
warning('Rounding ground truth bounding box data to integer values.')
end
boxes = round(boxes);
boxes(:,1:2) = max(boxes(:,1:2), 1);
boxes(:,3:4) = min(boxes(:,3:4), imageSize([2 1]));
end
end