BatchDatsetReader.py 文件源码

def _transform(self, filename, flag = False):
        if flag:
            image = np.array(Image.open(filename), dtype=np.uint8)
            image[image == 255] = 21
        else:
            image = misc.imread(filename)

        if self.__channels and len(image.shape) < 3:  # make sure images are of shape(h,w,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)