python loadimage

by · 2019年09月21日 · 256 Words · ~1min reading time | Improve on


原文链接： python loadimage
#定义图片数据集加载函数load_image

def load_image(data_dir): 
    labels = []
    images = []
       for d in directories:
           label_dir = os.path.join(data_dir, d)
           file_names = [os.path.join(label_dir, f)
                      for f in os.listdir(label_dir)
                      if f.endswith('.ppm')]
           for f in file_names:
            # 以指定的尺寸读取图像数据
               image_data = image.load_img(f, target_size=(config.img_rows, config.img_cols))
            # 将图片像素值转换为0-1
               image_data = image.img_to_array(image_data) / 255
               images.append(image_data)
               labels.append(int(d))
     return images,labels


 #加载训练集：
train_images, train_labels =load_data(train_data_dir)
#加载测试集：
test_images, test_labels =load_data(test_data_dir)

————————————————
版权声明：本文为CSDN博主「深度学习算法项目代码分享」的原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/weixin_37479258/article/details/101061670

Tags:

`