PSPNet tile_predict() error

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.38k stars 799 forks source link

PSPNet tile_predict() error #224

Open erichhhhho opened 5 years ago

erichhhhho commented 5 years ago

In here, we should average the overlapping prediction instead of replacing it with new psub Should replace pred[:, :, sx:ex, sy:ey] = psub with pred[:, :, sx:ex, sy:ey] += psub

As shown in here: