onetonfoot / unet

A Unet Implementation in PyTorch

Unet

A PyTorch implementation of Unet. The model can be found in unet.py and an example of usage can be found in unet.ipynb.