You could modify the read logic in __getitem__ and e.g. use a for loop to read all corresponding images.
In my current approach I’m just reading two files, but you could also read all matching file names.