Hello,
Are you able to change the number of output channels in the last layer? I am also trying to train deeplabv3 model with my dataset. I am following this link:

  
      
      My Journey with Deep Learning and Computer Vision – 30 Aug 19
  
  
    

Transfer Learning for Segmentation Using DeepLabv3 in PyTorch 10

In this post, I’ll be covering how to use a pre-trained semantic segmentation DeepLabv3 model for the task of road crack detection in PyTorch by using transfer learning. The same procedure ca…


  
  
    
    
  
  


But the result still shows 21 output channels. Can anyone please help me out with this? I have 3 class labels for my dataset.
If possible suggest some other github repo or a link to perform semantic segmentation.