Change the following line:-



 michelepy:

image_name = os.path.join(self.root_dir, self.labels.iloc[index, 0])


To:-
image_name = os.path.join(self.root_dir, str(self.labels.iloc[index, 0]))
and your label would be-
labels = self.labels.iloc[index, 1]
I hope this works out for you!
Feel free to text back!
Regards