README.md 1.81 KiB
Newer Older
Unknown's avatar
Unknown committed

DISCO Cell Localization CNN Scripts
=
This repository contains the relevant files for generation and visualization of the convolutional neural network (CNN) presented in the forthcoming publication indicated at the bottom of this page. 

Scripts
-
* kfold_hyperopt_binary_localizer.py - Uses the hyperopt package to explore and optimize the hyperparameter space of the CNN. 
* optimized_binary_localizer.py - Trains the final network with the optimized (and manually selected) hyperparameters. 

Notebooks
-
* genetic_optimization_visualizer.ipynb - Shows the results of the 'kfold_hyperopt_binary_localizer.py' script for selection of the optimal network hyperparameters. 
* training_log_plotter.ipynb - Plots the training history of the optimized network. 
* output_visualizer.ipynb - Loads the final network and shows segmentation predictions on validation data as well as the necessary downstream processing for laser vector path generation. 

Data
-
* binary_data.p -  The cell image data and annotations used as input to the network (training and validation).
* genetic_trials_cv.p - Output from 'kfold_hyperopt_binary_localizer.py' showing the optimization trials.
* binary_localizer_16_0.28892_1_54_7_12.log - The training log produced by 'optimized_binary_localizer.py'
* binary_localizer_16_0.28892_1_54_7_12.hdf5 - The final network used for automated cell laser lysis. 


**Note** these scripts were tested on:
* tensorflow==1.13.1
* hyperopt==0.2.3
* keras==2.2.4
* numpy==1.16.2
* opencv-python==3.3.0
* sklearn==0.20.1

Publication:
=
Lamanna, J.* , Scott, E.Y. * , Edwards, H.* , Chamberlain, M.D., Dryden, M.D.M., Peng, J., Mair, B., Lee, A., Sklavounos, A.A., Abbas, F., Moffat, J. & A.R. Wheeler. "Digital Microfluidic Isolation of Single Cells for - Omics". 2020.  
\* Co-first authors
DOI: (TBD)