I have a little problem regarding my code where I\'m trying to make pairs of test images for later comparison in few shot learning. My code is the following: