rishi8313 / pytorch-unet-resnet-50-encoder

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About


Languages

Language:Python 100.0%