Awesome

title: Water Body Segmentation emoji: 🤗 colorFrom: blue colorTo: gray sdk: gradio sdk_version: 3.10.0 app_file: app.py pinned: false

UNET Water Body Segmentation - PyTorch

This project contains the code for training and deploying a UNET model for water body segmentation from satellite images. The model is trained on the Satellite Images of Water Bodies from Kaggle. The model is trained using PyTorch and deployed using Gradio on Hugging Face Spaces.

🚀 Getting Started

All the code for training the model and exporting to ONNX format is present in the notebook folder or you can use this Kaggle Notebook for training the model. The app.py file contains the code for deploying the model using Gradio.

🤗 Demo

You can try out the model on Hugging Face Spaces

🖥️ Sample Inference