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
43 lines
1.3 KiB
Matlab
2 years ago
|
%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
|