.. image:: https://readthedocs.org/projects/tf-unet/badge/?version=latest :target: http://tf-unet.readthedocs.io/en/latest/?badge=latest :alt: Documentation Status

.. image:: http://img.shields.io/badge/arXiv-1609.09077-orange.svg?style=flat :target: http://arxiv.org/abs/1609.09077

.. image:: https://img.shields.io/badge/ascl-1611.002-blue.svg?colorB=262255 :target: http://ascl.net/1611.002

This is a generic U-Net implementation as proposed by Ronneberger et al. <https://arxiv.org/pdf/1505.04597.pdf>_ developed with Tensorflow. The code has been developed and used for Radio Frequency Interference mitigation using deep convolutional neural networks <http://arxiv.org/abs/1609.09077>_ .

The network can be trained to perform image segmentation on arbitrary imaging data. Checkout the Usage <http://tf-unet.readthedocs.io/en/latest/usage.html>_ section or the included Jupyter notebooks for a toy problem <https://github.com/jakeret/tf_unet/blob/master/demo/demo_toy_problem.ipynb>_ or the Radio Frequency Interference mitigation <https://github.com/jakeret/tf_unet/blob/master/demo/demo_radio_data.ipynb>_ discussed in our paper.

The code is not tied to a specific segmentation such that it can be used in a toy problem to detect circles in a noisy image.

.. image:: https://raw.githubusercontent.com/jakeret/tf_unet/master/docs/toy_problem.png :alt: Segmentation of a toy problem. :align: center

To more complex application such as the detection of radio frequency interference (RFI) in radio astronomy.

.. image:: https://raw.githubusercontent.com/jakeret/tf_unet/master/docs/rfi.png :alt: Segmentation of RFI in radio data. :align: center

Or to detect galaxies and star in wide field imaging data.

.. image:: https://raw.githubusercontent.com/jakeret/tf_unet/master/docs/galaxies.png :alt: Segmentation of a galaxies. :align: center