Downloads · 30 days
9
11% of all-time downloads
samarthkumar/ultrasound-nerve-segmentation
ultrasound-nerve-segmentation is a image segmentation model from samarthkumar. Use it for the image segmentation task on the model card, and read the license before you ship it in a product. It is set up for keras. The card lists the license as mit.
Binary segmentation of the Brachial Plexus nerve in ultrasound images.
Downloads · 30 days
9
11% of all-time downloads
All-time downloads
81
Public
Repo size
35.1 MB
Likes
0
Public
Click a slice to open those files.
.keras26.3 MB · 75%
From the Hugging Face model README
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
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))