forked from mindspore-Ecosystem/mindspore
!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:
commit
1605fd4a1e
|
@ -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
|
||||||
|
|
Loading…
Reference in New Issue