linbinbin92 / LSTM-UNet

LSTM-UNet

The code in this repository is suplamentary to our paper "Microscopy Cell Segmentation via Convolutional LSTM Networks" published in ISBI 2019. If this code is used please cite the paper:

@article{arbelleIsbi2019, title={Microscopy cell segmentation via convolutional LSTM networks}, author={Arbelle, Assaf and Raviv, Tammy Riklin}, booktitle={2019 IEEE 16th International Symposium on Biomedical Imaging (ISBI 2019)}, pages={1008--1012}, year={2019}, organization={IEEE} }

Getting Started

These instructions will get you a copy of the project up and running on your local machine for development and testing purposes. See deployment for notes on how to deploy the project on a live system.

Prerequisites

This project is writen in Python 3 and makes use of tensorflow 2.0.0a0. Please see the requierments.txt file for all prerequisits.

Installing

In order to get the code, either clone the project, or download a zip from GitHub:

git clone https://github.com/arbellea/LSTM-UNet.git

Install all python requierments

pip3 install -r <PATH TO REPOSITORY>/requirements.txt 

This should do it!

Data

The training script was tailored for the Cell Tracking Benchmarck If you do not have the training data and wish to train on the challenge data please contact the organizers through the website: www.celltrackingchallenge.net Once you have the data, untar the file metadata_file.tar.gz into the direcroty of the training data:

cd <PATH TO CELLTRACKINGCHALLENGE DATA>/Training
tar -xzvf  <PATH TO REPOSITORY>/metadata_files.tar.gz

make sure that metadata_01.pickle and metadata_02.pickle are located in each dataset directory (Only of 2D datasets)

Private Data:

If you would like to train on your private data, please create a metadata_.pickle file. Run the script './create_sequence_metadat.py' with the paths and formats for your data:

--root_dir : Root directory of sequence, example: '~/CellTrackingChallenge/Train/Fluo-N2DH-SIM+

--seq : Sequence number (two digit) , example: '01' or '02'

--raw_file_template : Template for image sequences, example: '01/t{:03d}.tif' where {:03d} will be replaced by a three digit number 000,001,...

-- seg_file_template : TemplateTemplate for image sequences segmentation , example: '01_GT/SEG/man_seg{:03d}.tif' where {:03d} will be replaced by a three digit number 000,001,...

--tra_file_template : Optional!. Template for image sequences tracking lables , example: '01_GT/TRA/man_track{:03d}.tif' where {:03d} will be replaced by a three digit number 000,001,...

--force : Force overwrite existing metadata pickle

Training

Modify Parameters

Open the Params.py file and change the paths for ROOT_DATA_DIR and ROOT_SAVE_DIR. ROOT_DATA_DIR should point to the directory of the cell tracking challenge training data: /Training and ROOT_SAVE_DIR should point to whichever directory you would like to save the checkpoints and tensorboard logs.

Run Training Script:

In order to set the parameters for training you could either change the parameters if Params.py file under class CTCParams or input them through command line. You are encourged to go over the parameters in CTCParams to see the avialable options The training script is train2D.py

python3 train2D.py

Training on Private Data:

Since there are many formats for data and many was to store annotations, we could note come up with a generic data reader. So if one would like to train on private data we recommend one of the following:

1. Save the data in the format of the cell tracking challenge and create the corresponding metadata_.pickle file.
2. Write you own Data reader with similar api to ours. See the data reader in DataHandling.py

Inference

Modify Parameters

Open the Params.py file and change the paths for ROOT_TEST_DATA_DIR. ROOT_TEST_DATA_DIR should point to the directory of the cell tracking challenge training data: /Test.

Download Pre-trained models for Cell Segmentation Benchmark

If you would like to run the pretrained models for the cell segmentation benchmark datasets, you could download the models from: https://drive.google.com/file/d/1uQOdelJoXrffmW_1OCu417nHKtQcH3DJ/view?usp=sharing

Run Inference Script:

In order to set the parameters for training you could either change the parameters if Params.py file under class CTCInferenceParams or input them through command line. You are encourged to go over the parameters in CTCInferenceParams to see the avialable options The training script is train2D.py

python3 Inference2D.py

Authors

Assaf Arbelle ([e-mail]) and Tammy Riklin Raviv ([e-mail]) Ben Gurion University of the Negev

License

This project is licensed under the MIT License - see the LICENSE.md file for details