文件夹自动标号分类

Posted 2020-11-09 ayew

def read_img(path):
    imgs = []
    labels = []
    # classs = []
    classs = os.listdir(path)

    for idx, folder in enumerate(classs):
        cate = os.path.join(path, folder)
        for im in os.listdir(cate):
            print(im)
            img_path =os.path.join(cate, im)
            print(‘reading the images:%s‘ % (img_path))
            img = io.imread(img_path)
            img = transform.resize(img, (w, h))
            # with open(‘tests.txt‘, ‘a‘) as f:
            #     f.write(img_path+‘_‘+str(idx)+‘\n‘)
            imgs.append(img)
            labels.append(idx)
    return np.asarray(imgs, np.float32), np.asarray(labels, np.int32)