bonlime / keras-deeplab-v3-plus

Keras implementation of Deeplab v3+ with pretrained weights
MIT License
1.36k stars 428 forks source link

mobilenetv2 #13

Closed bhack closed 6 years ago

bhack commented 6 years ago

Do you plan to add mobilenetv2 backbone like in https://github.com/tensorflow/models/blob/master/research/deeplab/model.py#L245?

bonlime commented 6 years ago

Right now I don't have enough time to add it, but will do it in June

bhack commented 6 years ago

Ok. I think that you can bootstrap from https://github.com/titu1994/MobileNetworks that is already loading official tensorflow weights. /cc @titu1994

Sucran commented 6 years ago

@bhack the newest keras version have mobilenetv2 in keras.application, you can check this https://github.com/keras-team/keras/blob/master/keras/applications/mobilenetv2.py

bhack commented 6 years ago

Hope to have this soon https://github.com/keras-team/keras/issues/9113

bonlime commented 6 years ago

@bhack
Had a day off, so added MobileNetv2 support earlier than expected. I checked that output is identical to original TF model Used model in keras.applications as a starting point

bhack commented 6 years ago

@bonlime Was it different from the official keras mobilenet?

bonlime commented 6 years ago

A little bit. Output Stride should 8, so in deeplab they don't use strides in last layers & add dilation rate when needed, to compensate this