Sky_Segmentation / README.md
Mohaaaa's picture
Upload 4 files
c44101e verified
---
title: Sky Segmentation App
emoji: 🌤️
colorFrom: blue
colorTo: purple
sdk: gradio
sdk_version: 4.44.0
app_file: app.py
pinned: false
license: mit
---
# Sky Segmentation App 🌤️
This is an interactive web application for sky segmentation using a trained U-Net model with ResNet50 encoder.
## Features
- 📁 Easy image upload interface
- 🎭 Real-time sky mask generation
- 🔵 Visual overlay showing detected sky regions
- 📱 Mobile-friendly responsive design
- ⚡ Fast inference on CPU/GPU
## How to Use
1. Upload an image (JPG, PNG, WEBP supported)
2. The model automatically processes your image
3. View three outputs:
- Original image
- Binary sky mask (white = sky)
- Colored overlay with sky highlighted in blue
## Model Details
- **Architecture**: U-Net with ResNet50 encoder
- **Framework**: PyTorch + Segmentation Models PyTorch
- **Input Size**: 512x512 pixels (automatically resized)
- **Output**: Binary segmentation mask
Built with ❤️ using Gradio and Hugging Face Spaces