← all repositories

zhixuhao/unet

A U-Net convolutional neural network for biomedical image segmentation implemented with Keras functional API.

4.9k stars Jupyter Notebook Computer VisionML Frameworks
unet
Velocity · 7d
+1.5
★ / day
Trend
steady
star history

This repository implements U-Net, a fully convolutional neural network architecture, for semantic image segmentation tasks. The model takes 512x512 input images and outputs pixel-level segmentation masks using a sigmoid activation function. It uses Keras functional API for flexible architecture experimentation and includes data augmentation via Keras ImageDataGenerator to compensate for limited training data. The implementation targets biomedical image segmentation, originally from the ISBI challenge.

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