Large-scale prediction using trained models

ELEKTRONN / elektronn3

A PyTorch-based library for working with 3D and 2D convolutional neural networks, with focus on semantic segmentation of volumetric biomedical image data

MIT License

161 stars 27 forks source link

Large-scale prediction using trained models #17

Closed mdraw closed 5 years ago

mdraw commented 6 years ago

Currently, only training scenarios are handled by elektronn3. Tools for large-scale model deployment on big data volumes will be needed eventually. This needs some discussion and careful planning first, because there are many open questions in this regard, e.g. w.r.t. if we should start working on multi-node distributed prediction in PyTorch or instead use another framework such as Caffe2 for this via ONNX export.

mdraw commented 5 years ago

This is now supported via the elektronn3.inference module.