networks

Pytorch implementation of benchmark models for image segmentation

• Models

  • UNet
  • UNet++
  • UNet3+
  • UNet3D
  • VNet

• Usage

    python setup.py develop
  

• Sample

  import torch
  from networks import UNet
  
  model = UNet(1,1)
  inputs = torch.randn((1,1,512,512))
  output = model(inputs)
  print(output.shape)