@ksz Thanks a lot for your precious help.
I try every day to learn more and more about pytorch, and recently I have discovered albumentations. From what I have read it contains some advantages when we want to do image augmentation for image segmentation for example, if I am not mistaken, as described here:

  
      
      albumentations.ai
  
  
    

Albumentations Documentation - Mask augmentation for segmentation 4

Albumentations: fast and flexible image augmentations


  
  
    
    
  
  


I want to combine albumentations and transforms because I wanted to know if it was possible and whether there are any benefits to combining them.
Once again thanks a lot for your answers.