I think you might lose values due to the applied interpolation in TF.rotate.
Could you set resample to PIL.Image.NEAREST for this method and check, if your output would contain all points?



 enterthevoidf22:

still i was confused how to perform validation split because i supply the random_split function with the custom dataset class that as i said already performs augmentation.


If you are lazily loading the data, just create two datasets. One with the training transformation, the other one with the validation transformation, and wrap both datasets in Subsets using the randomly split data indices.