thx, you really help me a lot. I override ImageFolder’s function init and getitem and achieve the goal.

volumeSet.png493×829 23.2 KB

But there are some new questions. After changing above part without changing dataloader, pycharm always warned me that ‘‘list index out of range’’. And if I set batch_size, I couldn’t get the expected output, such as the expected one is batch_size x 100x3x224x224, while what I got is like 100 x batch_size x 3x224x224.
So should I override dataloader? Could you give me some advece?