This tutorial 46 is quite old by now and written for Theano + Lasagne, however you could reuse the approach for the augmentation of the keypoints as it walks you through it step by step.



 enterthevoidf22:

how do i now pass this dataset together with a compose object and make sure that transforms are handling this dictionary correctly?


If you want to apply the same “random” transformation of two different image tensors, you could use the functional API of torchvision.transforms as described here 34.