Wrong dimension used in translation augmentation

[gathierry]

Hi,
When doing translation augmentation, I think x-translation should be computed with img.shape[1] while y-translation in the next line should be calculated with img.shape[0]

PyTorch_YOLOv4/utils/datasets.py

Line 644 in 1861063

T[0, 2] = random.uniform(-translate, translate) * img.shape[0] + border # x translation (pixels)

Please correct me if I got the wrong idea.