图像识别中的图片采样程序

图片库位于F:\study\img中，写一个采样程序，每次取10张图片。
一般训练神经网络时，不是将所有的图片一次性输入神经网络训练，而是分批次进行，那么就需要一个采样程序，如下：

import numpy as np
import os
import PIL.Image as img

path_image=r"F:\study\img"

class Sample:
    def read_data(self):
        self.img_arr=[]
        for name in os.listdir(path_image):
            imgs=img.open("{0}/{1}".format(path_image,name))
            imge=((np.array(imgs))/255-0.5)*2
            self.img_arr.append(imge)
        return self.img_arr

    def get_batch(self,set):
        self.get_arr=[]
        self.read_data()
        for i in range(set):
            num=np.random.randint(len(self.img_arr))
            image=self.img_arr[num]
            print(image)
            self.get_arr.append(image)
        return self.get_arr

sample=Sample()
sample.read_data()
sample.get_batch(10)

转载或引用请注明来源！