Solved by shubhvachher in post #3 
              
              
                Well, like @etekiller says, you can try to “Compose Transforms”. An example would be here. And then use image to PIL transform to solve your issue. This though will likely add a ton of time overhead to your data loading pipeline! 
If you’re not using many transforms from pytorch on your data during …