def zero_pad(image, target, h, w): oh, ow = get_tensor_image_size(target) pad_h = h - oh if oh < h else 0 pad_w = w - ow if ow < w else 0 image = F.pad(image, (0, 0, pad_w, pad_h), fill=0) target = F.pad(target, (0, 0, pad_w, pad_h), fill=255) return image, target
def __call__(self, img, target): """ Args: img (PIL Image): Image to be cropped. Returns: PIL Image: Cropped image. """ if self.padding > 0: img = F.pad(img, self.padding) target = F.pad(target, self.padding) i, j, h, w = self.get_params(img, self.size) return F.crop(img, i, j, h, w), F.crop(target, i, j, h, w),
def __call__(self, img): """ Args: img (numpy ndarray): Image to be padded. Returns: numpy ndarray: Padded image. """ return F.pad(img, self.padding, self.fill, self.padding_mode)
def __call__(self, img): """ Args: img (PIL Image): Image to be padded. Returns: PIL Image: Padded image. """ return F.pad(img, self.padding, self.fill)
def __call__(self, img): """ Args: img (PIL Image): Image to be cropped. Returns: PIL Image: Cropped image. """ if self.padding > 0: img = F.pad(img, self.padding) # pad the width if needed if self.pad_if_needed and img.size[0] < self.size[1]: img = F.pad(img, (int((1 + self.size[1] - img.size[0]) / 2), 0)) # pad the height if needed if self.pad_if_needed and img.size[1] < self.size[0]: img = F.pad(img, (0, int((1 + self.size[0] - img.size[1]) / 2))) i, j, h, w = self.get_params(img, self.size) return F.crop(img, i, j, h, w)
def __call__(self, img): """ Args: img (numpy ndarray): Image to be cropped. Returns: numpy ndarray: Cropped image. """ if self.padding is not None: img = F.pad(img, self.padding, self.fill, self.padding_mode) # pad the width if needed if self.pad_if_needed and img.shape[1] < self.size[1]: img = F.pad(img, (self.size[1] - img.shape[1], 0), self.fill, self.padding_mode) # pad the height if needed if self.pad_if_needed and img.shape[0] < self.size[0]: img = F.pad(img, (0, self.size[0] - img.shape[0]), self.fill, self.padding_mode) i, j, h, w = self.get_params(img, self.size) return F.crop(img, i, j, h, w)
def __call__(self, sample): """ Args: img (PIL Image): Image to be cropped. Returns: PIL Image: Cropped image. """ if self.padding > 0: sample = F.pad(sample, self.padding) (img, target) = sample i, j, h, w = self.get_params(img, self.size) return F.crop(sample, i, j, h, w)
def __call__(self, img): return F.pad(img, self.pad_width, self.mode)