码迷,mamicode.com
首页 > 其他好文 > 详细

加载图像并打乱顺序

时间:2020-04-14 16:46:38      阅读:58      评论:0      收藏:0      [点我收藏+]

标签:imp   bsp   mat   images   pen   文件路径   image   otl   标签   

 

import tensorflow as tf
import os
from matplotlib import pyplot as plt
import numpy as np
from sklearn.utils import shuffle
import skimage.io as io
def load_sample(sample_dir):
    lfilenames = []
    labelsnames = []
    for (dirpath, dirnames, filenames) in os.walk(sample_dir):#递归遍历文件夹
        for filename in filenames:                            #遍历所有文件名
            #print(dirnames)
            filename_path = os.sep.join([dirpath, filename])
            img=io.imread(filename_path);
            lfilenames.append(img)               #添加文件名
            labelsnames.append( dirpath.split(‘\\‘)[-1] )#添加文件名对应的标签
    return  shuffle(lfilenames,labelsnames)
data_dir = ‘F:\\python\\mnist_digits_images‘ #定义文件路径
image,label= load_sample(data_dir)   #载入文件名称与标签

 

  

加载图像并打乱顺序

标签:imp   bsp   mat   images   pen   文件路径   image   otl   标签   

原文地址:https://www.cnblogs.com/hsy1941/p/12698747.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!