Deeplab vs Erfnet vs Unet vs ResnetOC
Training UNet,ResnetOC, ErfNet and Deeplabv3+ on Mapillary dataset (1080)
Created on April 4|Last edited on April 4
Comment
Results
Получены результаты для четырёх моделей (Resnet OC, UNet, ErfNet и DeepLabV3+ с MobileNet в качестве бэкбона). Сети были обучены на 24(12) эпохах Mapillary.
Итого:
ErfNet mIOU ~ 15% fps ~ 18
UNet mIOU ~ 12% fps ~ 13
DeepLabV3+ mIOU ~ 23% fps ~ 20
Resnet OC mIOU ~ 26% fps ~ 27
Стоит учесть, что 24 эпохи - это значительно меньше, чем то количество эпох, которое использовали авторы. Кроме того, представленный IOU - это средний IOU на 66 классах. Ну и наконец - тренировка с размером батча 1-2 даёт свои негативные последствия.
DeepLab, например за 2 последние эпохи увеличил mIOU на 1%, т.е. если обучать дальше, качество скорее всего будет улучшаться.
Для тренировки DeepLab нужно минимум 15 гб памяти на одной карточке.
Resnet OC с предобученным бэкбоуном даёт лучший результат в принципе
Run set
5
Run set
5
Add a comment