search

facebookarchive / caffe2

Caffe2 is a lightweight, modular, and scalable deep learning framework.
https://caffe2.ai
Apache License 2.0
8.42k stars 1.94k forks source link

N-D data in Caffe2 #452

Open peterneher opened 7 years ago

peterneher commented 7 years ago

Hi everyone,

does Caffe2 support N-dimensional datasets? For example 3D medical image data with multiple contrsats (4th dimension) and multiple time steps (5th dimension). So at least 3D convolutions, and all the other corresponding oprations (pooling, deconvolution, ...), are necessary.

Cheers, Peter

akyrola commented 7 years ago

Yes it does. For example we have ConvNd in python/cnn.py.