milesial/Pytorch-UNet

This is stars and forks stats for /milesial/Pytorch-UNet repository. As of 02 May, 2024 this repository has 7429 stars and 2221 forks.

U-Net: Semantic segmentation with PyTorch Customized implementation of the U-Net in PyTorch for Kaggle's Carvana Image Masking Challenge from high definition images. Quick start Without Docker With Docker Description Usage Docker Training Prediction Weights & Biases Pretrained model Data Quick start Without Docker Install CUDA Install PyTorch 1.13 or later Install dependencies pip install -r requirements.txt Download the data and run training: bash scripts/download_data.sh python train.py --amp With...