UNet++ Bridge Semantic Segmentation
Weights for a UNet++ model segmenting bridge inspection images into:
- background
- beton (concrete)
- steel
Original Space: https://huggingface.co/spaces/Suzyloubna/Bridge.Semantic.Segmentation
Usage (Python)
from huggingface_hub import hf_hub_download
import torch
REPO_ID = "Suzyloubna/bridge-unetpp"
FILENAME = "MILESTONE_090_ACHIEVED_iou_0.9077.pth" # adjust if you renamed
weights_path = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
state = torch.load(weights_path, map_location="cpu")
Notes
- File is large; first download is cached.
- Consider converting to
safetensorsfor extra safety/performance.
License
Apache-2.0