Python图像处理【10】基于离散余弦变换的图像压缩

基于离散余弦变换的图像压缩

0. 前言

由于图像中相邻像素间的相关性引起的空间冗余、图像序列中不同帧之间存在相关性引起的时间冗余，因此我们需要对图像数据进行压缩。数据压缩的目的就是通过去除这些数据冗余来减少数据表示所占用的存储空间。随着大数据时代的到来，图像数据在质量提高的同时，其大小也变得越来越大，在存储、传输、处理时非常困难，因此图像数据的压缩就显得非常重要。
图像压缩是数据压缩技术在图像上的应用，利用图像压缩可以减少图像数据中的冗余信息，从而能够更加高效存储和传输数据。在本节中，我们首先介绍离散余弦变换 (

Discrete Cosine Transform

DCT

) 的基本原理，利用

DCT

执行图像压缩操作，并实现图像

JPEG

压缩。

1. 离散余弦变换基础

离散余弦变换 (

Discrete Cosine Transform

DCT

) 以不同频率振幅的余弦函数组合表示图像，而离散傅里叶变换 (

Discrete Fourier Transform

DFT

) 则以复杂指数的形式使用正弦和余弦函数。以下方程式显示了如何计算

DCT

的第

    (
   
   
    i
   
   
    ,
   
   
    j
   
   
    )
   
  
  
   (i,j)
  
 
(i,j) 项(系数)：

 
  
   
    
     D
    
    
     (
    
    
     i
    
    
     ,
    
    
     j
    
    
     )
    
    
     =
    
    
     
      1
     
     
      
       
        2
       
       
        N
       
      
     
    
    
     C
    
    
     (
    
    
     i
    
    
     )
    
    
     C
    
    
     (
    
    
     j
    
    
     )
    
    
     
      ∑
     
     
      
       x
      
      
       =
      
      
       0
      
     
     
      
       N
      
      
       −
      
      
       1
      
     
    
    
     
      ∑
     
     
      
       y
      
      
       =
      
      
       0
      
     
     
      
       N
      
      
       −
      
      
       1
      
     
    
    
     f
    
    
     (
    
    
     x
    
    
     ,
    
    
     y
    
    
     )
    
    
     c
    
    
     o
    
    
     s
    
    
     [
    
    
     
      
       (
      
      
       2
      
      
       x
      
      
       +
      
      
       1
      
      
       )
      
      
       i
      
      
       π
      
     
     
      
       2
      
      
       N
      
     
    
    
     ]
    
    
     c
    
    
     o
    
    
     s
    
    
     [
    
    
     
      
       (
      
      
       2
      
      
       y
      
      
       +
      
      
       1
      
      
       )
      
      
       i
      
      
       π
      
     
     
      
       2
      
      
       N
      
     
    
    
     ]
    
   
   
     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
   
   
    )
   
   
    =
   
   
    
     {
    
    
     
      
       
        
         
          1
         
         
          
           2
          
         
        
       
      
      
       
        
       
      
      
       
        
         
          u
         
         
          =
         
         
          0
         
        
       
      
     
     
      
       
        
         1
        
       
      
      
       
        
       
      
      
       
        
         
          u
         
         
          >
         
         
          0
         
        
       
      
     
    
   
  
  
   C(u)=\left\{ \begin{array}{rcl} \frac 1 {\sqrt 2} & & {u=0}\\ 1 & & {u>0} \end{array} \right.
  
 
C(u)={211u=0u>0，对于块大小为 

 
  
   
    8
   
   
    ×
   
   
    8
   
  
  
   8\times8
  
 
8×8 的

DCT

，

2. 基于离散余弦变换的图像压缩

DCT

算法用途广泛，通常用于实现图像

JPEG

压缩，在本节中，我们将学习如何使用

Python

执行图像重建操作。

(1) 首先，导入所需库，关键在于从

scipy.fftpack

模块导入所需的函数

dct()

和

idct()

：

import scipy.fftpack as fp
from skimage.io import imread
from skimage.color import rgb2gray, gray2rgb
from skimage.draw import rectangle_perimeter
import numpy as np
import matplotlib.pylab as plt
from mpl_toolkits.mplot3d import Axes3D  # noqa: F401 unused importfrom mpl_toolkits.axes_grid1 import make_axes_locatable
from matplotlib.ticker import LinearLocator, FormatStrFormatter
from 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

图像块的

个

DCT

基函数，对于图像中的每个块，

个

DCT

系数是基函数的线性组合以生成图像块系数：

DCT 系数

接下来，我们将实现简化版

JPEG

压缩，以确保仅需少数

DCT

系数可以表示图像，而不会导致图像质量有过多的视觉失真。在本节中，我们将直接在输入图像像素上计算

8x8

图像块的

DCT

。此外，我们将使用全局阈值来清零大部分

DCT

系数，而不是通过用量化矩阵(对于给定的压缩级别)分割

DCT

矩阵来进行量化。最后，我们将跳过霍夫曼 (

Huffman

) 编码，并存储(压缩)/读取(解压缩)图像。

3.2 JPEG 压缩实践

(1) 导入所需库，读取输入图像，并将图像分割为

8x8

的

DCT

块：

import scipy.fftpack as fp
from skimage.io import imread
from skimage.color import rgb2gray, gray2rgb
from skimage.draw import rectangle_perimeter
import numpy as np
import matplotlib.pylab as plt
from mpl_toolkits.mplot3d import Axes3D  # noqa: F401 unused importfrom mpl_toolkits.axes_grid1 import make_axes_locatable
from matplotlib.ticker import LinearLocator, FormatStrFormatter
from scipy.fftpack import dct, idct

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')

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 =112
plt.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

系数置为

，打印应用阈值后剩余的非零

DCT

系数的百分比。

thresh =0.03
dct_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()

DCT 系数
(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】使用高通滤波器执行边缘检测

标签： python 图像处理人工智能

本文转载自: https://blog.csdn.net/qq_30167691/article/details/128434681
版权归原作者 AI technophile 所有，如有侵权，请联系我们删除。

Python图像处理【10】基于离散余弦变换的图像压缩

基于离散余弦变换的图像压缩

0. 前言

1. 离散余弦变换基础

2. 基于离散余弦变换的图像压缩

3. 图像 JPEG 压缩

3.1 JPEG 压缩原理

3.2 JPEG 压缩实践

小结

系列链接

发表评论

“Python图像处理【10】基于离散余弦变换的图像压缩”的评论:

关于作者

overfit同步小助手

相关阅读

文章导航