Sorry I’m new to pytorch, what do you mean by mapping?
I put my code link below if you are interested.

  
      
      GitHub
  
  
    

xysong1201/image_segmentation 76

Contribute to xysong1201/image_segmentation development by creating an account on GitHub.