speedinghzl / pytorch-segmentation-toolbox

PyTorch Implementations for DeeplabV3 and PSPNet
MIT License
768 stars 167 forks source link

How to train the deeplab v3? #13

Closed MrLinNing closed 5 years ago

MrLinNing commented 5 years ago

Can you provide more details about training deeplab v3 on cityscapes dataset? Besides the mIoU=78.30 about the PSPnet is the valid result, and how about the test results of cityscapes dataset? Can you give the detail of training it? Thank you! @speedinghzl

speedinghzl commented 5 years ago

@MrLinNing Our reproduced PSPNet can achieve ~78.5 on val set and ~77.5 on test set using single scale testing. You can run Deeplabv3 by replacing this line and keeping the hyperparameters the same. After the conference deadline, I will reorganize the code and support more models and datasets. Thanks.