deJQK:

If you finish it, will you make it public like on github? That will be beneficial to many others.


Yes, I’ll will. I ran into this issue with PIL last week, trying to work with multi-channel images. Replacing the TorchVision’s dependency on PIL sounds like the right thing to do. It looks like TorchVision library already supports multiple imaging backends, so it should be a matter of adding a cv2 backend.

  
      
      GitHub
  
  
    

pytorch/vision 160

vision - Datasets, Transforms and Models specific to Computer Vision


  
  
    
    
  
  


I’ll let you know once I’m done.