Solved by Oli in post #2 
              
              
                Helooo  
The best way would be to create your own dataset class. In it you could return your images (or whatever data you have) plus your index for that data item. Try this and report back if you have a hard time creating your dataset