基于离散余弦变换的图像压缩
0. 前言
由于图像中相邻像素间的相关性引起的空间冗余、图像序列中不同帧之间存在相关性引起的时间冗余，因此我们需要对图像数据进行压缩。数据压缩的目的就是通过去除这些数据冗余来减少数据表示所占用的存储空间。随着大数据时代的到来，图像数据在质量提高的同时，其大小也变得越来越大，在存储、传输、处理时非常困难，因此图像数据的压缩就显得非常重要。
图像压缩是数据压缩技术在图像上的应用，利用图像压缩可以减少图像数据中的冗余信息，从而能够更加高效存储和传输数据。在本节中，我们首先介绍离散余弦变换 (
Discrete Cosine Transform
,
DCT
) 的基本原理，利用
DCT
执行图像压缩操作，并实现图像
JPEG
压缩。
1. 离散余弦变换基础
离散余弦变换 (
Discrete Cosine Transform
,
DCT
) 以不同频率振幅的余弦函数组合表示图像，而离散傅里叶变换 (
Discrete Fourier Transform
,
DFT
) 则以复杂指数的形式使用正弦和余弦函数。以下方程式显示了如何计算
N×NN×NN×N 图像f(x,y)f(x,y)f(x,y) 的
DCT
的第
(i,j)(i,j)(i,j) 项(系数)：D(i,j)=12NC(i)C(j)∑x=0N−1∑y=0N−1f(x,y)cos[(2x+1)iπ2N]cos[(2y+1)iπ2N]D(i,j)=\frac 1 {\sqrt{2N}}C(i)C(j)\sum ^{N-1} _{x=0}\sum ^{N-1} _{y=0}f(x,y)cos[\frac {(2x+1)i\pi}{2N}]cos[\frac {(2y+1)i\pi}{2N}]D(i,j)=2N1C(i)C(j)x=0∑N−1y=0∑N−1f(x,y)cos[2N(2x+1)iπ]cos[2N(2y+1)iπ]
其中
C(u)={12u=01u>0C(u)=\left\{ \begin{array}{rcl} \frac 1 {\sqrt 2} & & {u=0}\\ 1 & & {u>0} \end{array} \right.C(u)={211u=0u>0，对于块大小为8×88\times88×8 的
DCT
，
N=8N=8N=8 。
2. 基于离散余弦变换的图像压缩
DCT
算法用途广泛，通常用于实现图像
JPEG
压缩，在本节中，我们将学习如何使用
Python
执行图像重建操作。
(1) 首先，导入所需库，关键在于从
scipy.fftpack
模块导入所需的函数
dct()
和
idct()
：
import scipy.fftpack as fpfrom skimage.io import imreadfrom skimage.color import rgb2gray, gray2rgbfrom skimage.draw import rectangle_perimeterimport numpy as npimport matplotlib.pylab as pltfrom mpl_toolkits.mplot3d import Axes3D # noqa: F401 unused importfrom mpl_toolkits.axes_grid1 import make_axes_locatablefrom matplotlib.ticker import LinearLocator, FormatStrFormatterfrom scipy.fftpack import dct, idct
(2) 使用
scipy.fftpack
的
dct()
和
idct()
函数分别定义函数
dct2()
和
idct2()
以实现
2D-DCT
和
IDCT
。其中，函数
dct()
的调用方式如下，而
idct()
函数的调用方式类似：
scipy.fftpack.dct(x,type=2, n=None, axis=-1, norm=None, overwrite_x=False)
该返回任意类型序列x的离散余弦变换。
根据在前言中介绍的
DCT
方程，并使用通过使用参数
norm ='ortho'
应用归一化：
defdct2(a):return dct(dct(a, axis=0, norm='ortho'), axis=1, norm='ortho')defidct2(a):return idct(idct(a, axis=0, norm='ortho'), axis=1, norm='ortho')
(3) 读取输入
RGB
图像并将其转换为灰度。计算图像的
2D-DCT
，然后利用
IDCT
使用前面定义的函数重建图像：
im = rgb2gray(imread('1.png'))imF = dct2(im)im1 = idct2(imF)
(4) 检查重建的图像，观察其与原始图像之间的差别，查看其是否与原始图像相同，在这里使用函数
np.callclose()
评估图像间的差距：
print(np.allclose(im, im1))
(5) 使用
matplotlib.pyplot
绘制原始图像和重建图像，得到结果如下所示：
plt.figure(figsize=(10,5))plt.gray()plt.subplot(121), plt.imshow(im), plt.axis('off'), plt.title('original image', size=10)plt.subplot(122), plt.imshow(im1), plt.axis('off'), plt.title('reconstructed image (DCT+IDCT)', size=10)plt.tight_layout()plt.show()
在以上结果图像中可以看到，重建后的图像与原始图像在视觉效果上类似，证明了可以使用
DCT
进行图像压缩，使用
IDCT
对压缩后的图像进行重建。
3. 图像 JPEG 压缩
在上一小节中，我们已经学习了如何实现
2D-DCT
和
IDCT
，在本节中，我们将使用这两种技术执行图像压缩操作，完成
JPEG
格式图像压缩。
3.1 JPEG 压缩原理
JPEG
压缩基本操作步骤如下所示：
- 将图像分割为
8x8的图像块 - 在每个图像块上(从左到右，从上到下)应用
DCT - 对每个图像块使用量化压缩
- 将压缩后的图像块存储在较小的空间中，从而达到较高的压缩效果
下图显示了在
8x8
图像块的
64
个
DCT
基函数，对于图像中的每个块，
64
个
DCT
系数是基函数的线性组合以生成图像块系数：
接下来，我们将实现简化版
JPEG
压缩，以确保仅需少数
DCT
系数可以表示图像，而不会导致图像质量有过多的视觉失真。在本节中，我们将直接在输入图像像素上计算
8x8
图像块的
DCT
。此外，我们将使用全局阈值来清零大部分
DCT
系数，而不是通过用量化矩阵(对于给定的压缩级别)分割
DCT
矩阵来进行量化。最后，我们将跳过霍夫曼 (
Huffman
) 编码，并存储(压缩)/读取(解压缩)图像。
3.2 JPEG 压缩实践
(1) 导入所需库，读取输入图像，并将图像分割为
8x8
的
DCT
块：
import scipy.fftpack as fpfrom skimage.io import imreadfrom skimage.color import rgb2gray, gray2rgbfrom skimage.draw import rectangle_perimeterimport numpy as npimport matplotlib.pylab as pltfrom mpl_toolkits.mplot3d import Axes3D # noqa: F401 unused importfrom mpl_toolkits.axes_grid1 import make_axes_locatablefrom matplotlib.ticker import LinearLocator, FormatStrFormatterfrom scipy.fftpack import dct, idctdefdct2(a):return dct(dct(a, axis=0, norm='ortho'), axis=1, norm='ortho')defidct2(a):return idct(idct(a, axis=0, norm='ortho'), axis=1, norm='ortho')im = rgb2gray(imread('1.png'))dct_coeffs = np.zeros(im.shape)for i inrange(0, im.shape[0],8):for j inrange(0, im.shape[1],8):dct_coeffs[i:(i+8),j:(j+8)]= dct2(im[i:(i+8),j:(j+8)])
(2) 从图像中提取一个图像块，并显示该块的
DCT
系数。
index =112plt.figure(figsize=(10,6))plt.gray()plt.subplot(121), plt.imshow(im[index:index+8,index:index+8]), plt.title("An 8x8 Image block", size=10)plt.subplot(122), plt.imshow(dct_coeffs[index:index+8,index:index+8], vmax= np.max(dct_coeffs)*0.01, vmin =0, extent=[0, np.pi, np.pi,0])plt.title("An 8x8 DCT block", size=10)plt.show()
输出结果如下所示：
(3) 使用全局阈值对
DCT
系数进行阈值设置，将值小于阈值乘以最大系数值的所有
DCT
系数置为
0
，打印应用阈值后剩余的非零
DCT
系数的百分比。
thresh =0.03dct_thresh = dct_coeffs *(abs(dct_coeffs)>(thresh*np.max(dct_coeffs)))percent_nonzeros = np.sum( dct_thresh !=0.0)/(im.shape[0]*im.shape[1])print("Keeping only {}% of the DCT coefficients".format(percent_nonzeros*100.0))# Keeping only 3.45875% of the DCT coefficients
从以上输出可以看出，应用阈值后，只有约
3.5％
的
DCT
系数为非零值。
(4) 可视化原始
DCT
系数以及阈值处理后获得的系数：
plt.figure(figsize=(12,7))plt.gray()plt.subplot(121), plt.imshow(dct_coeffs,cmap='gray',vmax = np.max(dct_coeffs)*0.01,vmin =0), plt.axis('off')plt.title("8x8 DCTs of the image", size=10)plt.subplot(122), plt.imshow(dct_thresh, vmax = np.max(dct_coeffs)*0.01, vmin =0), plt.axis('off')plt.title("Thresholded 8x8 DCTs of the image", size=10)plt.tight_layout()plt.show()
(5) 最后，仅使用
3.5％
非零
DCT
系数，使用
8x8
块
IDCT
重建压缩图像：
im_out = np.zeros(im.shape)for i inrange(0, im.shape[0],8):for j inrange(0, im.shape[1],8):im_out[i:(i+8),j:(j+8)]= idct2( dct_thresh[i:(i+8),j:(j+8)])
(6) 绘制原始图像和
DCT
压缩图像，并对比原始图像和重建(压缩)图像：
plt.figure(figsize=(15,7))plt.gray()plt.subplot(121), plt.imshow(im), plt.axis('off'), plt.title('original image', size=10)plt.subplot(122), plt.imshow(im_out), plt.axis('off'), plt.title('DCT compressed image', size=10)plt.tight_layout()plt.show()
从以上图像可以看到，仅通过有效地存储大约
3.5％
的
DCT
系数，我们便可以高图像质量的重建图像，而不会过多的降低图像质量上，从而实现较高压缩比，并计算
MSE/PSNR
以进行定量测量图像间的差距。
小结
离散余弦变换 (
Discrete Cosine Transform
,
DCT
) 以不同频率振幅的余弦函数组合表示图像，而离散傅里叶变换 (
Discrete Fourier Transform
,
DFT
) 则以复杂指数的形式使用正弦和余弦函数。在实际应用中，
DCT
算法用途广泛，通常用于实现图像
JPEG
压缩，在本节中，我们学习了如何利用
DCT
执行图像
JPEG
压缩，用于更加高效存储和传输图像数据。
系列链接
Python图像处理【1】图像与视频处理基础
Python图像处理【2】探索Python图像处理库
Python图像处理【3】Python图像处理库应用
Python图像处理【4】图像线性变换
Python图像处理【5】图像扭曲/逆扭曲
Python图像处理【6】通过哈希查找重复和类似的图像
Python图像处理【7】采样、卷积与离散傅里叶变换
Python图像处理【8】使用低通滤波器模糊图像
Python图像处理【9】使用高通滤波器执行边缘检测
版权归原作者 AI technophile 所有， 如有侵权，请联系我们删除。