'caffe2 - create own MNIST lmdb dataset from numpy array

I am currently trying to create a MNIST lmdb dataset from my own data saved as numpy arrays data (shape (50000,1,28,28)) and label (shape (50000,)).

Does someone know how to convert numpy arrays to a lmdb dataset suitable for a caffe2 model?

KR ulk



Sources

This article follows the attribution requirements of Stack Overflow and is licensed under CC BY-SA 3.0.

Source: Stack Overflow

Solution Source