Solved by hrushi in post #4 
              
              
                I’m not really sure whether it can store metadata, but what I generally do is create two numpy arrays, one for images and other for labels. If I want the label for i-th image, I simply do label[i].