GitHub - bonlime/keras-deeplab-v3-plus: Keras implementation of Deeplab v3+ with...
source link: https://github.com/bonlime/keras-deeplab-v3-plus
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.
README.md
keras-deeplab-v3-plus
Keras implementation of Deeplab v.3 Model is based on the original TF frozen graph. It is possible to load pretrained weights into this model. Weights are directly imported from original TF checkpoint
Segmentation results of original TF model. Output Stride = 8
This result is obtained as an argmax applied to logits at exit layer
Segmentation results of this repo model with loaded weights and OS = 8
Results are identical to the TF model
Segmentation results of this repo model with loaded weights and OS = 16 Results are still good
How to use this model with custom input shape and custom number of classes:from model import Deeplabv3
deeplab_model = Deeplabv3((512,512,3), num_classes=4, last_activation=True, OS=8)
deeplab_model.load_weights('deeplabv3_weights_tf_dim_ordering_tf_kernels.h5', by_name = True)
You can find deeplabv3_weights_tf_dim_ordering_tf_kernels.h5 file in releases
Recommend
About Joyk
Aggregate valuable and interesting links.
Joyk means Joy of geeK