← all repositories

ethanhe42/u-net

A Keras implementation of the U-Net convolutional neural network for biomedical image segmentation tasks like ultrasound nerve detection.

451 stars Python Computer VisionML Frameworks
u-net
Velocity · 7d
+0.1
★ / day
Trend
steady
star history

This repository provides a tutorial implementation of the U-Net architecture for the Kaggle Ultrasound Nerve Segmentation competition. The model is a convolutional auto-encoder with skip connections between encoder and decoder layers at corresponding levels, implemented using Keras with TensorFlow backend. It processes ultrasound images resized to 64x80 pixels and outputs segmentation masks scaled between 0 and 1.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.