如何使用将本地图片封装成一个四维矩阵

    def __init__(self,image_options,path):
        self.image_options = {'resize': True, 'resize_size': IMAGE_SIZE}
        self.path="/data0/user/data/fenlei/datadir"

    def _read_images(self):
        self.__channels = True

        self.all_path=[]
        self.solupath=[]
        self.label = []

        # 得到模型下所有的数据绝对路径
        directories = [x[0] for x in os.walk(self.path)]

        #directories[0]代表的是主文件路劲,[1:]代表的是子文件夹路径
        self.label=[]
        num=0
        for label,directory in enumerate(directories[1:]):
            class_num= [os.path.join(label) for label in os.listdir(directory)]

            self.all_path = [os.path.join(directory,label) for label in class_num]
            self.solupath.extend(self.all_path)
            for i,j in enumerate(self.all_path):
                self.label.append(num)
            num+=1
        self.images=np.array([self._transform(img) for img in self.solupath])
        return self.images, np.array(self.label)

    def _transform(self,filename):
        image = misc.imread(filename)
        if self.__channels and len(image.shape)<3:
            image = np.array([image for i in range(3)])

        if self.image_options.get("resize", False) and self.image_options["resize"]:
            resize_size = int(self.image_options["resize_size"])
            resize_image = misc.imresize(image,
                                         [resize_size, resize_size], interp='nearest')
        else:
            resize_image = image

        return np.array(resize_image)

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值