search

Tramac / Lightweight-Segmentation

Lightweight models for real-time semantic segmentation(include mobilenetv1-v3, shufflenetv1-v2, igcv3, efficientnet).
Apache License 2.0
356 stars 77 forks source link

wrong inter_channels in mobilenetv3_seg.py #7

Open nieyan opened 5 years ago

nieyan commented 5 years ago

https://github.com/Tramac/Lightweight-Segmentation/blob/71bf91f3d5b946a046becfd5d8667e8a0f80d630/light/model/mobilenetv3_seg.py#L17 It should be: inter_channels = 112 if mode == 'large' else 24

by the way, mode == 'small' might be wrong too