Skip to content

Convolutional LSTM for video segmentation with Keras

Notifications You must be signed in to change notification settings

farquasar1/ConvLSTM

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 

Repository files navigation

ConvLSTM

Implementation of a Convolutional LSTM with Keras for video segmentation.

Data should be provided in separate folder as a set of videos (mp4 format) and the corresponding segmentation mask with the suffix _label in the filename (before extension).

To train the model run script

lstm_train_fcn.py 

It will load the data, compress the resolution by a factor a 4 - the shape of input should be (W,D,C), respectively (96,108,1). Here I'm considering only 1 channel, i.e., black and white images. For colour images change the shape.

Training should take 1 hour per video sequence of 1000 frames in an NVIDIA TitanX

Example of ultrasound video sequence and corresponding image segmentation with Convolutional LSTM (middle) and only convolutions (top)

About

Convolutional LSTM for video segmentation with Keras

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages