Use the picture name to label the picture [tips]

problem

When doing the second classification task , There are two types of pictures in a folder . Originally, two types of pictures are more common in different folders , This situation requires another way of thinking .

solve

Use the picture name to judge , The code snippet is as follows ：

import os
path = ' Write the address of the picture folder here '
photos = os.listdir(path)# Put all picture names together, including suffixes 
Y = []
for ph in photos:
    x = os.path.splitext(ph)[0]# Separate picture names and suffixes , Pick up the name 
    # The number in the string of the picture name is （ Or use letters isalpha()） Or another way to equal a certain character '{}'.format(x[-1])=='e' To make a judgment 
    if (x[-1]).isdigit():
        Y.append(1)
    else:
        Y.append(0)