Data_understanding

list_images.py

#Put all the images into a list
listOfImageNames = []
for dirname, _, filenames in os.walk('/kaggle/input/insulation-joint-training-set-prorail/trainset_insulation_joint/images/'):
    for filename in filenames:
        #print(os.path.join(dirname, filename))
        listOfImageNames.append(os.path.join(dirname, filename))

#define a function that  display an image and it's label knowing the indice of this image
def display_image(indice):
  print(df[df['label'].index==indice]) #print the label of the image 
  display(Image(filename=listOfImageNames[indice], width=700, height=700))