Mentions légales du service

Skip to content
Snippets Groups Projects
Select Git revision
  • master default protected
  • v1.0
2 results
Name Last commit Last update
bin
deepfinder
docs
examples
images
pyqt
test/utils
LICENSE
README.md
requirements.txt
requirements_gpu.txt
README.md

Deep Finder

The code in this repository is described in this pre-print. This paper has been submitted to Nature Methods and has now been published.

Disclaimer: A more recent version of DeepFinder can be found here: https://github.com/deep-finder/cryoet-deepfinder/

Contents

System requirements

Deep Finder has been implemented using Python 3 and is based on the Keras package. It has been tested on Linux (Debian 10), and should also work on Mac OSX as well as Windows.

The algorithm needs an Nvidia GPU and CUDA to run at reasonable speed (in particular for training). The present code has been tested on Tesla K80 and M40 GPUs. For running on other GPUs, some parameter values (e.g. patch and batch sizes) may need to be changed to adapt to available memory.

- If above conditions are not met, we cannot guarantee the functionality of our code at this time.

Package dependencies

Deep Finder depends on following packages. The package versions for which our software has been tested are displayed in brackets:

tensorflow-gpu (2.6.0)
keras          (2.6.0)
numpy          (1.19.5)
h5py           (3.1.0)
lxml           (4.3.4)
scikit-learn   (0.21.2)     
scikit-image   (0.15.0)  
matplotlib     (3.1.0)
mrcfile        (1.1.2)
PyQt5          (5.13.2)
pyqtgraph      (0.10.0)

Installation guide

Before installation, you need a python environment on your machine. If this is not the case, we advise installing Miniconda.

Then, you need to download the present repository. Next, open a terminal, place yourself in your deep-finder folder and run:

cd /path/to/deep-finder/
pip install -r requirements_gpu.txt

Also, in order for Keras to work with your Nvidia GPU, you need to install CUDA. For more details about installing Keras and CUDA, please see Keras installation instructions.

Once these steps have been achieved, the user should be able to run Deep Finder.

Instructions for use

Using the scripts

Instructions for using Deep Finder are contained in folder examples/. The scripts contain comments on how the toolbox should be used. To run a script, first place yourself in its folder. For example, to run the target generation script:

cd examples/training/
python step1_generate_target.py

Using the GUI

The GUI (Graphical User Interface) is launchable from folder bin/, and should be more intuitive for those who are not used to work with script. Currently, 5 GUIs are available (tomogram annotation, target generation, training, segmentation, clustering) and allow the same functionalities as the scripts in example/. To run a GUI, first open a terminal. For example, to run the target generation GUI:

/path/to/deepfinder/bin/generate_target

Training GUI

Notes:

  • working examples are contained in examples/analyze/, where Deep Finder processes the test tomogram from the SHREC'19 challenge.
  • The script in examples/training/ will fail because the training data is not included in this Gitlab.
  • The evaluation script (examples/analyze/step3_launch_evaluation.py) is the one used in SHREC'19, which needs additional packages (pathlib and pycm, can be installed with pip). The performance of Deep Finder has been evaluated by an independent group, and the result of this evaluation has been published in Gubins & al., "SHREC'19 track: Classification in cryo-electron tomograms".