Semantic Segmentation using deeplabv3+resnet101 from torchvision models

Thank you for your help. Extracting class weights using the mean frequency balancing method and adding it to crossentropyloss solved my problem,

Thanks again for your help

1 Like