!6448 [MS][CPU][DataAugmentation][IMP] Fix for call to PadBBoxes in RandomCropWithBBox Op

Merge pull request !6448 from danishnxt/AugOpsFix
This commit is contained in:
mindspore-ci-bot 2020-09-18 09:49:21 +08:00 committed by Gitee
commit 1605fd4a1e
1 changed files with 2 additions and 2 deletions

View File

@ -46,8 +46,8 @@ Status RandomCropWithBBoxOp::Compute(const TensorRow &input, TensorRow *output)
&padded_image_w, &padded_image_h, &crop_further)); &padded_image_w, &padded_image_h, &crop_further));
// update bounding boxes with new values based on relevant image padding // update bounding boxes with new values based on relevant image padding
if (t_pad_left || t_pad_bottom) { if (t_pad_left || t_pad_top) {
RETURN_IF_NOT_OK(BoundingBox::PadBBoxes(&(*output)[1], boxCount, t_pad_left, t_pad_top)); RETURN_IF_NOT_OK(BoundingBox::PadBBoxes(&(*output)[1], boxCount, t_pad_top, t_pad_left));
} }
if (!crop_further) { if (!crop_further) {
// no further cropping required // no further cropping required