Skip to main content
Help us improve PyPI by participating in user testing. All experience levels needed!

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.

Logo+Example

Membrane and mitochondria probability maps. Predicted with a CNN with recursive training. Data: zebra finch area X dataset j0126 by Jörgen Kornfeld.

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

This version
History Node

1.0.14

History Node

1.0.13

History Node

1.0.12

History Node

1.0.11

History Node

1.0.10

History Node

1.0.9

History Node

1.0.8

History Node

1.0.7

History Node

1.0.6

History Node

1.0.5

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Filename, size & hash SHA256 hash help File type Python version Upload date
elektronn-1.0.14-cp27-cp27m-manylinux1_x86_64.whl (484.1 kB) Copy SHA256 hash SHA256 Wheel cp27 Apr 14, 2017
elektronn-1.0.14-cp27-cp27mu-manylinux1_x86_64.whl (484.1 kB) Copy SHA256 hash SHA256 Wheel cp27 Apr 14, 2017
elektronn-1.0.14.tar.gz (109.0 kB) Copy SHA256 hash SHA256 Source None Apr 14, 2017

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging CloudAMQP CloudAMQP RabbitMQ AWS AWS Cloud computing Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page