samarthkumar/ultrasound-nerve-segmentation

Ultrasound Nerve Segmentation — U-Net

Binary segmentation of the Brachial Plexus nerve in ultrasound images.

Model: U-Net (TensorFlow/Keras) Dataset: Kaggle Ultrasound Nerve Segmentation Validation Accuracy: 98.47% Input: 128×128 grayscale ultrasound image Output: 128×128 binary segmentation mask

Usage

from huggingface_hub import hf_hub_download
from src.model import load_trained_model
import numpy as np

model_path = hf_hub_download(repo_id="samarthkumar/ultrasound-nerve-segmentation", filename="model_nerve.keras")
model = load_trained_model(model_path)

# image shape: (128, 128, 1), values in [0, 1]
mask = model.predict(np.expand_dims(image, 0))