My implemenetation of a modular Unet from the paper "U-Net: Convolutional Networks for Biomedical Image Segmentation"
- Lucidrains
- Agorians
pip install unet-torch
import torch
from unet.model import Unet
model = Unet(n_channels=3, n_classes=1)
inputs = torch.randn(1, 3, 256, 256)
outputs = model(inputs)
print(outputs.size)
print(outputs)
MIT
@misc{1505.04597,
Author = {Olaf Ronneberger and Philipp Fischer and Thomas Brox},
Title = {U-Net: Convolutional Networks for Biomedical Image Segmentation},
Year = {2015},
Eprint = {arXiv:1505.04597},
}