U-Net
=====

.. currentmodule:: torchgeo.models
.. autofunction:: unet
.. autoclass:: Unet_Weights