Трансформации в PyTorch. Как сохранить синхронизацию с target, если target это координаты пикселя на изображении?

Я создают нейросеть в Pytorch, которая должна искать на изображении объекты (отвёрнутые от камеры лица). По факту мне нужны 4 координаты, для примера прилагаю изображение на котором помечены эти координаты. В pytorch есть хороший набор трансформаторов, но проблема в том, что если трансформация связана с искажением изображения, то надо соответственно менять и target, иначе координаты не будут совпадать с нужным точкам. Как лучше это сделать? Возможно в PyTorch уже есть, реализованные для таких целей инструменты? А вот так выглядит target для этого изображения - это nympy массив

[[1253  368]
 [1251  589]
 [1386  579]
 [1368  800]]