nyud_layers.py 文件源码

def load_image(self, idx):
        """
        Load input image and preprocess:
    - resize
        - cast to float
        - switch channels RGB -> BGR
        - subtract mean
        - transpose to channel x height x width order
        """
        idx = self.indices[idx]
    idx = idx.split()[0]
    im = io.imread('{}/{}'.format(self.data_dir, idx))
        im = Image.fromarray(im)
        im = im.resize((self.width, self.height), Image.ANTIALIAS)   # resize image
        im = np.array(im, dtype=np.float32)              # cast to float
        im = im[:,:,::-1]                                            # RGB -> BGR
        im -= self.mean_bgr                      # mean subtraction
        im = im.transpose((2,0,1))
        return im