The goal of this repository is to provide an extendable toolbox for large-scale experimentation with deep learning techniques for image enhancement and segmentation of N-dimensional biological imaging data. It is designed to support high-performance computing clusters and utilize GPU-acceleration.
Training datasets, as well as model checkpoints used in the paper will be made accessible soon!
Run the following -->
conda create -n raygun python=3.9 tensorflow pytorch torchvision torchaudio cudatoolkit=11.3 affogato -c pytorch -c nvidia -c conda-forge
conda activate raygun
pip install git+https://github.com/htem/raygun
Should you run into gcc / boost errors when conda/pip installing raygun, try installing libboost
first:
sudo apt-get update
sudo apt-get install libboost-all-dev
raygun-train path/to/train_config.json