How to calculate the loss for pspnet?

meetps / pytorch-semseg

Semantic Segmentation Architectures Implemented in PyTorch

https://meetshah.dev/semantic-segmentation/deep-learning/pytorch/visdom/2017/06/01/semantic-segmentation-over-the-years.html

MIT License

3.39k stars 796 forks source link

How to calculate the loss for pspnet? #195

Open UcanSee opened 5 years ago

UcanSee commented 5 years ago

I want to train pspnet on my own dataset, but I don't know how to calculate the loss because there are two outputs x and x_aux when training, the x_aux is not same size with the input, how should i calculate the loss?