Fix Dimension Mismatch in CutMix Augmentation

[GummyJum]

Description

This pull request addresses a bug in the CutMix augmentation implementation where the cut_width was incorrectly calculated based on image_height instead of image_width.

• Motivation: Correct bounding box dimensions in CutMix by using image_width to calculate cut_width.
• Context: The current code calculates both cut_height and cut_width using image_height.
• Change: Update cut_width calculation to use image_width.

Type of change

• [x] Bug fix (non-breaking change which fixes an issue)

Checklist

• [x] I have signed the Contributor License Agreement.
• [x] I have read guidelines for pull request.
• [x] My code follows the coding guidelines.
• [x] I have performed a self code review of my own code.
• [x] My changes generate no new warnings.

[google-cla[bot]]

Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

View this failed invocation of the CLA check for more information.

For the most up to date status, view the checks section at the bottom of the pull request.