Hi, I have a tricky problem (at least to me) and am not sure how to proceed.
Let’s say I have a dataset of images and I have generated some labels for every batch. These are stored in batches of size b_size
How this goes for b_size = 32
:
- Traverse dataset and generate batches of size 32 so something like
(32, 1, 64, 64)
. - For every batch I have a set of labels of size
(32, 100, 1, 1)
I want to create a dataset/dictionary that can store these image batches and corresponding labels so that when I later iterate over this dataset/dictionary it returns the (x,y)
pair where x is the batch of images
and y is the corresponding batch of labels
I would prefer a dataset since I also want to subset it in the future. But I honestly have no idea how to start this.
Any ideas would be appreciated thanks!