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

Merge pull request !6448 from danishnxt/AugOpsFix
5 years ago · 1605fd4a1e
parent 980059eaec 2473adf24f
commit 1605fd4a1e
1 changed files with 2 additions and 2 deletions
--- a/mindspore/ccsrc/minddata/dataset/kernels/image/random_crop_with_bbox_op.cc
+++ b/mindspore/ccsrc/minddata/dataset/kernels/image/random_crop_with_bbox_op.cc
@ -46,8 +46,8 @@ Status RandomCropWithBBoxOp::Compute(const TensorRow &input, TensorRow *output)
                               &padded_image_w, &padded_image_h, &crop_further));

  // update bounding boxes with new values based on relevant image padding
-  if (t_pad_left || t_pad_bottom) {
-    RETURN_IF_NOT_OK(BoundingBox::PadBBoxes(&(*output)[1], boxCount, t_pad_left, t_pad_top));
+  if (t_pad_left || t_pad_top) {
+    RETURN_IF_NOT_OK(BoundingBox::PadBBoxes(&(*output)[1], boxCount, t_pad_top, t_pad_left));
  }
  if (!crop_further) {
    // no further cropping required