/pytorch-unet

UNet implementation in pytorch

Primary LanguagePython

U-Net PyTorch Implementation

This repository contains a PyTorch implementation of the U-Net architecture for semantic segmentation on the Oxford Pets Dataset. The implementation is based on the original paper by Ronneberger et al. [1].

Citation

@article{ronneberger2015u,
  title={U-net: Convolutional networks for biomedical image segmentation},
  author={Ronneberger, Olaf and Fischer, Philipp and Brox, Thomas},
  journal={International Conference on Medical image computing and computer-assisted intervention},
  pages={234--241},
  year={2015},
  publisher={Springer}
}