create model erfnet_resa

voldemortX / pytorch-auto-drive

PytorchAutoDrive: Segmentation models (ERFNet, ENet, DeepLab, FCN...) and Lane detection models (SCNN, RESA, LSTR, LaneATT, BézierLaneNet...) based on PyTorch with fast training, visualization, benchmarking & deployment help

BSD 3-Clause "New" or "Revised" License

847 stars 138 forks source link

create model erfnet_resa #74

Closed LittleJohnKhan closed 2 years ago

LittleJohnKhan commented 2 years ago

ERFNetEncoder is copied from utils/models/segmentation/erfnet.py, then add pretrained-weights initialization

voldemortX commented 2 years ago

LGTM!

[ ] Try use DDP
[x] Try SCNN style ERFNet
[x] Tune lr on validation set

LittleJohnKhan commented 2 years ago

erfnet_v2 with ERFNet as backbone