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 Distributions
Hashes for elektronn-1.0.14-cp27-cp27mu-manylinux1_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 32c59c0b73a9a681022fd12c254deb8e5f18fc8c4e34de02a4872b46c9c9529f |
|
MD5 | 0c60f86cff2be8b73c60a188a7567d09 |
|
BLAKE2b-256 | 5528bb7ce874fec46f5e2f413dedcad23086f865154ba635f60e55c511a959f1 |
Hashes for elektronn-1.0.14-cp27-cp27m-manylinux1_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | c857207792e5ca19f6702ba4ed8af62a865eaf9658b38ec40fe0c51739054016 |
|
MD5 | e28c0cb5610ae82c269582f050507d29 |
|
BLAKE2b-256 | d9dbd6a6124a20e517cdf69c7294ef5f318bf5bfb6598c4667707842b0206f62 |