A highly configurable toolkit for training 3d/2d CNNs and general Neural Networks
Project description
ELEKTRONN is a highly configurable toolkit for training 3D/2D CNNs and general Neural Networks.
It is written in Python 2 and based on Theano, which allows CUDA-enabled GPUs to significantly accelerate the pipeline.
The package includes a sophisticated training pipeline designed for classification/localisation tasks on 3D/2D images. Additionally, the toolkit offers training routines for tasks on non-image data.
ELEKTRONN was created by Marius Killinger and Gregor Urban at the Max Planck Institute For Medical Research to solve connectomics tasks.
Membrane and mitochondria probability maps. Predicted with a CNN with recursive training. Data: zebra finch area X dataset j0126 by Jörgen Kornfeld.
Learn More:
Toy Example
$ elektronn-train MNIST_CNN_warp_config.py
This will download the MNIST data set and run a training defined in an example config file. The plots are saved to ~/CNN_Training/2D/MNIST_example_warp.
File structure
ELEKTRONN ├── doc # Documentation source files ├── elektronn │ ├── examples # Example scripts and config files │ ├── net # Neural network library code │ ├── scripts # Training script and profiling script │ ├── training # Training library code │ └── ... ├── LICENSE.rst ├── README.rst └── ...
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for elektronn-1.0.12-cp27-cp27m-manylinux1_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | e5f41530ab831926d935706dbfbb80481e27cce03297aebb2c2a179122c0d5c5 |
|
MD5 | f515e84ca441dd0799e9743078733b51 |
|
BLAKE2b-256 | 5284286248c3789a957088b945032bec270599de47db511ddc5ea3f02b17f339 |