We try to reproduce the paper: "Automated Pavement Crack Segmentation Using U-Net-Based Convolutional Neural Network" using only PyTorch.
Make sure that when installing PyTorch to have a good look at: https://pytorch.org/get-started/locally/ to enable GPU if you have CUDA.