Нужна помощь в понимании этого кода Pytorch-Vision, чтобы я мог улучшить качество обучения сети

Question

Я работаю над проектом, использующим набор данных из этой бумаги , и вот ссылка github на код, используемый для обучения и генерации весов

My вопрос относится к файловой утилите / transforms.py

Метод RandomHorizontFlip - это то, что я пытаюсь понять

class RandomHorizontalFlip(object):
    def __init__(self, prob):
        self.prob = prob

    def __call__(self, image, target):
        if random.random() < self.prob:
            height, width = image.shape[-2:]
            image = image.flip(-1)
            bbox = target["boxes"]
            bbox[:, [0, 2]] = width - bbox[:, [2, 0]]
            target["boxes"] = bbox
            if "masks" in target:
                target["masks"] = target["masks"].flip(-1)
            if "keypoints" in target:
                keypoints = target["keypoints"]
                keypoints = _flip_coco_person_keypoints(keypoints, width)
                target["keypoints"] = keypoints
        return image, target

Как бы я go реализовал вертикальное отражение или любое другое преобразование , что я должен иметь в виду ??