PyTorch-based human image segmentation pipeline using a U-Net model with an EfficientNet-B0 encoder, segmentation-safe augmentations, and binary mask prediction.
python opencv computer-vision deep-learning cnn pytorch transfer-learning image-segmentation unet semantic-segmentation efficientnet albumentations binary-segmentation segmentation-models-pytorch segmentation-dataset
-
Updated
Jul 28, 2026 - Jupyter Notebook