Spaces:
Runtime error
Runtime error
| 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](https://www.kaggle.com/datasets/franciscoescobar/satellite-images-of-water-bodies) from Kaggle. The model is trained using PyTorch and deployed using [Gradio](https://gradio.app/) on [Hugging Face Spaces](https://huggingface.co/spaces). | |
| ## 🚀 Getting Started | |
| All the code for training the model and exporting to ONNX format is present in the [notebook](notebooks) folder or you can use this [Kaggle Notebook](https://www.kaggle.com/code/gauthamkrishnan119/water-body-segmentation-pytorch) for training the model. The [app.py](app.py) file contains the code for deploying the model using Gradio. | |
| ## 🤗 Demo | |
| You can try out the model on [Hugging Face Spaces](https://huggingface.co/spaces/gauthamk/water-body-segmentation) | |
| ## 🖥️ Sample Inference | |
|  | |