yassouali / pytorch-segmentation

:art: Semantic segmentation models, datasets and losses implemented in PyTorch.
MIT License
1.68k stars 381 forks source link

Deeplabv3 #113

Closed IzzaRabbani closed 3 years ago

IzzaRabbani commented 3 years ago

what need to be changed to train on deeplabv3?

yassouali commented 3 years ago

Hi @IzzaRabbani

Check out the pending pull request that proposes a possible correction, I still need to check it out but it looks like a possible fix (the backbone used), also see issue https://github.com/yassouali/pytorch-segmentation/issues/78 and other for more details.