Skip to content

Latest commit

 

History

History
5 lines (3 loc) · 442 Bytes

README.md

File metadata and controls

5 lines (3 loc) · 442 Bytes

pytorch-unet-resnet-50-encoder

This model is a U-Net with a pretrained Resnet50 encoder. For most segmentation tasks that I've encountered using a pretrained encoder yields better results than training everything from scratch, though extracting the bottleneck layer from the PyTorch's implementation of Resnet is a bit of hassle so hopefully this will help someone!

You will need PyTorch version >= 0.3.0 and TorchVision version >= 0.2.0