| 35 | |
| 36 | |
| 37 | Data Augmentations::Resize(Data mData, int width, int height, float probability) { |
| 38 | float rand_number = RandomNum<float>(0, 1); |
| 39 | if (rand_number <= probability) { |
| 40 | |
| 41 | float h_scale = height * 1.0 / mData.image.rows; |
| 42 | float w_scale = width * 1.0 / mData.image.cols; |
| 43 | for (int i = 0; i < mData.bboxes.size(); i++) |
| 44 | { |
| 45 | mData.bboxes[i].xmin = int(w_scale*mData.bboxes[i].xmin); |
| 46 | mData.bboxes[i].xmax = int(w_scale*mData.bboxes[i].xmax); |
| 47 | mData.bboxes[i].ymin = int(h_scale*mData.bboxes[i].ymin); |
| 48 | mData.bboxes[i].ymax = int(h_scale*mData.bboxes[i].ymax); |
| 49 | } |
| 50 | |
| 51 | cv::resize(mData.image, mData.image, cv::Size(width, height)); |
| 52 | |
| 53 | } |
| 54 | return mData; |
| 55 | } |