Skip to content

Latest commit

 

History

History
2 lines (2 loc) · 350 Bytes

File metadata and controls

2 lines (2 loc) · 350 Bytes

UNet-Semantic-Segmentation-Football

A PyTorch implementation of U-Net architecture to perform pixel-level semantic segmentation on the Football Scene Dataset. Includes custom mask preprocessing, robust data augmentation with Albumentations, and fine-tuning with Class Weighted CrossEntropyLoss to resolve data imbalance for small object detection.