CoolFace
Apppublic

ruturajmane/road_segmentation_unet

sourceHugging Facemitupdated 6mo agoView on Hugging Face
1likes
App README

Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

Road Segmentation UNet

This model performs road segmentation on satellite imagery using UNet architecture.

Model Details

  • —Architecture: UNet
  • —Framework: PyTorch
  • —Input Size: 256x256 (change if different)
  • —Output: Binary mask (road / background)

Files

  • —bestunetroad_segmentation.pth → trained weights

How to Use

python
import torch
from model import UNet   # your model file

model = UNet()
model.load_state_dict(torch.load("best_unet_road_segmentation.pth", map_location="cpu"))
model.eval()