Creating and training a U-Net model with PyTorch for 2D & 3D semantic segmentation: Dataset building [1/4]
A guide to semantic segmentation with PyTorch and the U-Net
Published in
10 min readDec 2, 2020
In this series (4 parts) we will perform semantic segmentation on images using plain PyTorch and the U-Net architecture. I will cover the following topics: