def resize(self, ratio):
"""Return a resized copy of the original SRImage by the given resize ratio.
@param ratio: resize ratio
@type ratio: float
@return: a resized copy of the original SRImage
@rtype: L{sr_image.SRImage}
"""
size = sr_image_util.create_size(self._size, ratio)
resized_image = self._image.resize(size, Image.BILINEAR)
return SRImage(resized_image)
def resize(self, ratio):
"""Return a resized copy of the original SRImage by the given resize ratio.
@param ratio: resize ratio
@type ratio: float
@return: a resized copy of the original SRImage
@rtype: L{sr_image.SRImage}
"""
size = sr_image_util.create_size(self.size, ratio)
resized_image = sr_image_util.resize(self._y_data, size)
return SRImage(resized_image, self._cb_data, self._cr_data)
def resize(self, ratio):
"""Return a resized copy of the original SRImage by the given resize ratio.
@param ratio: resize ratio
@type ratio: float
@return: a resized copy of the original SRImage
@rtype: L{sr_image.SRImage}
"""
size = sr_image_util.create_size(self.size, ratio)
resized_image = sr_image_util.resize(self._y_data, size)
return SRImage(resized_image, self._cb_data, self._cr_data)
def _downgrade(self, ratio, kernel):
"""Downgrade the original SR image with the given ratio and blur kernel.
@param ratio: downgrade ratio
@type ratio: float
@param kernel: blur kernel
@type kernel: L{PIL.ImageFilter.Kernel}
@return: downgraded image with same size as original image
@rtype: L{sr_image.SRImage}
"""
size = sr_image_util.create_size(self._size, ratio)
blured_image = self._image.filter(kernel)
downgraded_image = blured_image.resize(size, Image.BILINEAR)
return SRImage(downgraded_image.resize(self._size, Image.BILINEAR))
def _downgrade(self, ratio, kernel):
"""Downgrade the original SR image with the given ratio and blur kernel.
@param ratio: downgrade ratio
@type ratio: float
@param kernel: blur kernel
@type kernel: L{numpy.array}
@return: downgraded image with same size as original image
@rtype: L{sr_image.SRImage}
"""
size = sr_image_util.create_size(self.size, 1.0 / ratio)
blurred_image = sr_image_util.filter(self._y_data, kernel)
downgraded_image = sr_image_util.resize(blurred_image, size)
downgraded_image = sr_image_util.resize(downgraded_image, self._size)
return SRImage(downgraded_image, self._cb_data, self._cr_data)
def _downgrade(self, ratio, kernel):
"""Downgrade the original SR image with the given ratio and blur kernel.
@param ratio: downgrade ratio
@type ratio: float
@param kernel: blur kernel
@type kernel: L{numpy.array}
@return: downgraded image with same size as original image
@rtype: L{sr_image.SRImage}
"""
size = sr_image_util.create_size(self.size, 1.0/ratio)
blurred_image = sr_image_util.filter(self._y_data, kernel)
downgraded_image = sr_image_util.resize(blurred_image, size)
downgraded_image = sr_image_util.resize(downgraded_image, self._size)
return SRImage(downgraded_image, self._cb_data, self._cr_data)
def test_create_size(self):
self.assertEqual([1, 1], sr_image_util.create_size([1, 1], 1.0))
self.assertEqual([125, 125], sr_image_util.create_size([100, 100],
1.25))
self.assertEqual([50, 50], sr_image_util.create_size([100, 100], 0.5))
def test_create_size(self):
self.assertEqual([1, 1], sr_image_util.create_size([1, 1], 1.0))
self.assertEqual([125, 125], sr_image_util.create_size([100, 100], 1.25))
self.assertEqual([50, 50], sr_image_util.create_size([100, 100], 0.5))
