/squeeze-unet

Squeeze-unet Semantic Segmentation for embedded devices

Primary LanguagePython

Squeeze-unet Semantic Segmentation for embedded devices

The model is inspired by Squeezenet and U-Net.

camvid, is used as a Dataset.

Example of predicted image.

Requirements

  • Keras 2

About network

As a typical U-Net architecture, it has encoder and decoder parts, which consist of fire modules proposed by squeezenet.

TODO

  • Report speed vs accuracy.
  • link for download camvid dataset converted by cvision.