Hi,
It is really hard to say what the exact problem is but it seems data loader is generating indices that is out of bound for your lists. I think changing __len__ can help. Try this:

    def __len__(self):
        return len(self.new_images)

Also, can you let me know the shape of new_image and new_labels?
Bests