代码分析

对于图片读取的代码进行分析

首先，需要找到数据集当前的路径，写入path里，并定义一个读取分配的函数

#path=？
#读取图片
def get_files(path):
    '''
    Args:
        file_dir: file directory
    Returns:
        list of images and labels
    '''
    daisy=[]
    dandelion=[]
    roses=[]
    sunflowers=[]
    tulips=[]
    label_daisy=[]
    label_dandelion=[]
    label_roses=[]
    label_sunflowers=[]
    label_tulips=[]

对数据集进行规整的处理，将他们分配到制定的属性下，分配完毕后，将归纳的花类输出出来。

    for file in os.listdir(path):
        print file
        if file=='daisy':
            for files in os.listdir(path+file+'/'):
                daisy.append(path+file+'/'+files)
                label_daisy.append(0)           
        elif file=='roses':
            for files in os.listdir(path+file+'/'):
                roses.append(path+file+'/'+files)
                label_roses.append(1) 
               
        elif file=='sunflowers':
            for files in os.listdir(path+file+'/'):
                sunflowers.append(path+file+'/'+files)
                label_sunflowers.append(2) 
        elif file=='dandelion':
            for files in os.listdir(path+file+'/'):
                dandelion.append(path+file+'/'+files)
                label_dandelion.append(3) 
               
        else:
             for files in os.listdir(path+file+'/'):
                 tulips.append(path+file+'/'+files)
                 label_tulips.append(4) 

这样，数据就可以按照我们想要的形式处理好了。