If I wanted to write my own custom implementations of these classes, which methods would I need to implement?
A lot of torchvision.transforms
are a simple class with a __call__
method as seen here. You could thus implement a custom class using this template or you could alternatively also use transforms.Lambda
, which would accept your custom function.
alright thank you, will try these out