Hello,
I want to train a multi-stream CNN model with pytorch. My data looks like this:
Two folder train1 and train2, under these two folders, there are 2 classes, 1 and 2, in each class there are bunch of images. The name of the pictures under train1 and train2 are same.
train1:
----- 1:
a1.jpg b1.jpg c1.jpg d1.jpg
------2:
a2.jpg b2.jpg c2.jpg d2.jpg
train2:
----- 1:
a1.jpg b1.jpg c1.jpg d1.jpg
------2:
a2.jpg b2.jpg c2.jpg d2.jpg
For pytorch, how can I input two images from train1 and train2 simutaneously. For example, if we set batch size as 1:
train1/1/a1.jpg and train2/1/a1.jpg should be returned by the data loader at the same time.
How can I implement this, or any other ideas to do this any help will be appreciated
Thanks