def save(self, path, extension): if self._cb_data is not None and self._cr_data is not None: self._cb_data = sr_image_util.resize(self._cb_data, self._size) self._cr_data = sr_image_util.resize(self._cr_data, self._size) image = sr_image_util.compose(self._y_data, self._cb_data, self._cr_data) image.save(path, extension)
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 downgrade(self, size, kernel): """Downgraded the image to given size and blurred with given kernel. @param size: target downgrade size @param kernel: blur kernel @return: downgraded image @rtype: L{SRImage} """ blurred_image = sr_image_util.filter(self._y_data, kernel) downgraded_image = sr_image_util.resize(blurred_image, size) return SRImage(downgraded_image, self._cb_data, self._cr_data)
def upgrade(self, size, kernel): """Upgrade the image to the given size and blurred with given kernel. @param size: target upgrade size @param kernel: blur kernel @return: upgraded image @rtype: L{SRImage} """ upgraded_image = sr_image_util.resize(self._y_data, size) blurred_image = sr_image_util.filter(upgraded_image, kernel) return SRImage(blurred_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)