search

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
160 stars 27 forks source link

Support 2D data sets #5

Closed mdraw closed 5 years ago

mdraw commented 6 years ago

Both the data loading (PatchCreator etc.) and the training components (StoppableTrainer) currently only support 3D (volumetric) image data sets. Support for 2D images would be nice, as long as it doesn't interfere with 3D support. In many parts of the code, 2D support could be transparently added as a special case of 3D without code duplication.

mdraw commented 5 years ago

Basic support for 2D images and an example data set for 2D training are available since #22. There are currently no plans for better built-in 2D data loading features, but contributions in this regard are welcome.