Top AI Repos — open-source AI, indexed and scored
Top AI Repos tracks AI repositories on GitHub and answers two different questions about each one: is it moving right now, and would you bet a product on it.
Top AI Repos tracks AI repositories on GitHub and answers two different questions about each one: is it moving right now, and would you bet a product on it.
Deep Learning Tutorial for Kaggle Ultrasound Nerve Segmentation competition, using Keras
| Date | Stars |
|---|---|
| 2026-07-31 | 944 |
| 2026-08-04 | 944 |
| 2026-08-06 | 944 |
Today
— stars today
This week
— stars this week
This month
— stars this month
Momentum
0.0
growth rate 0.00%/day
# Deep Learning Tutorial for Kaggle Ultrasound Nerve Segmentation competition, using Keras
This tutorial shows how to use [Keras library](http://keras.io/) to build deep neural network for ultrasound image nerve segmentation.
More info on this Kaggle competition can be found on [https://www.kaggle.com/c/ultrasound-nerve-segmentation](https://www.kaggle.com/c/ultrasound-nerve-segmentation).
This deep neural network achieves **~0.57 score on the leaderboard** based on test images,
and can be a good staring point for further, more serious approaches.
The architecture was inspired by [U-Net: Convolutional Networks for Biomedical Image Segmentation](http://lmb.informatik.uni-freiburg.de/people/ronneber/u-net/).
---
## Update 02.04.2017.
Changes:
- using Keras 2
- using TF backend instead of Theano
- using scikit-image instead of cv2
- added code for saving predicted images to disk
- training data is now split into train/validation data (80%/20%)
---
## Overview
### Data
[Provided data](https://www.kaggle.com/c/ultrasound-nerve-segmentation/data) is processed by ```data.py``` script.
This script just loads the images and saves them into NumPy binary format files **.npy** for faster loading later.
### Pre-processing
The images are not pre-processed in any way, except resizing to 96 x 96. Since the images are pretty noisy,
I expect that some thoughtful pre-processing could yield better performance of the model.
Output images (masks) are scaled to \[0, 1\] interval.
### Model
The provided model is basically a convolutional auto-encoder, but with a twist - it has skip connections from encoder layers to decoder layers that are on the same "level".
See picture below (note that image size and numbers of convolutional filters in this tutorial differs from the original U-Net architecture).
This deep neural network is implemented with Keras functional API, which makes it extremely easy to experiment with different interesting architectures.
Output from the network is a 96 x 96 which represents mask that should be learned. Sigmoid activation function
makes sure that mask pixels are in \[0, 1\] range.
### Training
The model is trained for 20 epochs, where each epoch took ~30 seconds on Titan X. Memory footprint of the model is ~800MB.
After 20 epochs, calculated Dice coefficient is ~0.68, which yielded ~0.57 score on leaderboard, so obviously this model overfits (cross-validation pull requests anyone? ;)).
Loss function for the training is basically just a **negative of Dice coefficient**
(which is used as [evaluation metric on the competition](https://www.kaggle.com/c/ultrasound-nerve-segmentation/details/evaluation)),
and this is implemented as custom loss function using Keras backend - check ```dice_coef()``` and ```dice_coef_loss()``` functions in ```train.py``` for more detail.
Also, for making the loss function smooth, a factor ```smooth = 1``` factor is added.
The weights are updated by Adam optimizer, with a 1e-5 learning rate. During training, model's weights are saved in HDF5 format.
---
## How to use
### Dependencies
This tutorial depends on the following libraries:
* scikit-image
* Tensorflow
* Keras >= 2.0
Also, this code should be compatible with Python versions 2.7-3.5.
### Prepare the data
In order to extract raw images and save them to *.npy* files,
you should first prepare its structure. Make sure that ```raw``` dir is located in the root of this project.
Also, the tree of ```raw``` dir must be like:
```
-raw
|
---- train
| |
| ---- 1_1.tif
| |
| ---- …
|
---- test
|
---- 1.tif
|
---- …
```
* Now run ```python data.py```.
Running this script will create train and test images and save them to **.npy** files.
### Define the model
* Check out ```get_unet()``` in ```train.py``` to modify the model, optimizer and loss function.
### Train the model and generate masks for test images
* Run ```pExcerpt of 5,379 characters
Read on GitHub
12
1
Would you bet a product on this? Bounded 0–100 and slow moving.
matched fp:2f32f3cdca4fa6a4, desc:tutorial