p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

PadBboxes Call fixed in RandomCropWithBboxOp

This commit is contained in:
Danish Farid 2020-09-17 18:28:41 -04:00
parent 8c7cc7943d
commit 2473adf24f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/minddata/dataset/kernels/image/random_crop_with_bbox_op.cc
View File

@ -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