def _apply_mask(self, mask):
if self._th > self._h:
mask = F.pad(mask, (self._th - self._h, 0), self.padding_maskvalue)
if self._tw > self._w:
mask = F.pad(mask, (0, self._tw - self._w), self.padding_maskvalue)
return mask[self._y:self._y + self._th, self._x:self._x + self._tw]
def _apply_mask(self, mask):
return F.pad(mask, self.size, self.mask_value)
def _apply_image(self, image):
if self._th > self._h:
image = F.pad(image, (self._th - self._h, 0), self.padding_value)
if self._tw > self._w:
image = F.pad(image, (0, self._tw - self._w), self.padding_value)
return image[self._y:self._y + self._th, self._x:self._x + self._tw]
def _apply_image(self, image):
return F.pad(image, self.size, self.value)
