An open API service indexing awesome lists of open source software.

https://github.com/fel-thomas/simple_unet

Simple Unet implementation to copy paste
https://github.com/fel-thomas/simple_unet

deep-learning image-segmentation machine-learning python satellite-imagery tensorflow u-net unet

Last synced: about 1 year ago
JSON representation

Simple Unet implementation to copy paste

Awesome Lists containing this project

README

          



Unet to copy paste


Simple implementation of Unet top copy paste, using Keras and Tensorflow (2.X).
The implementation is offered in two forms : plain file and as a function.

Original paper [U-Net: Convolutional Networks for Biomedical Image Segmentation](https://arxiv.org/abs/1505.04597).

![Example of Unet architecture](assets/unet_example.png)

---
## Quick Start

```python
model = build_Unet(input_shape=(128, 128, 3), output_channels=1, depth=4)
model.summary()
```
### Dependencies

* Tensorflow (2.X)

You need Python 3.5 or later.

## Acknowledgements

- [@tensorflow](https://github.com/tensorflow/tensorflow)
- [@keras](https://github.com/keras-team/keras)
```
"U-Net: Convolutional Networks for Biomedical Image Segmentation."
Olaf Ronneberger, Philipp Fischer, and Thomas Brox
arXiv preprint arXiv:1505. 04597, 2015.
```

🖐 Thomas FEL, 2020