《机器学习》—— SVD奇异值分解方法对图像进行压缩

news2024/9/24 20:25:02

文章目录

一、SVD奇异值分解简单介绍
二、代码实现—SVD奇异值分解方法对图像进行压缩

一、SVD奇异值分解简单介绍

SVD（奇异值分解）是一种在信号处理、统计学、线性代数、机器学习等多个领域广泛应用的矩阵分解方法。它将任何 m×n 矩阵 A 分解为三个特定矩阵的乘积：
在这里插入图片描述

其中：
- U 是一个 m×m 的正交矩阵（或 m×min(m,n)，取决于是否计算完整矩阵），其列被称为左奇异向量。
- Σ 是一个 m×n 的矩形对角矩阵（但通常以 m×min(m,n) 的形式出现，并且只在对角线上有值，其余位置为0），对角线上的元素称为奇异值，它们是非负的，并按降序排列。在实际应用中，Σ 经常以一维数组的形式返回，仅包含对角线上的奇异值。
- V^T 是 V 的转置，V 是一个 n×n 的正交矩阵（或 n×min(m,n)），其列被称为右奇异向量。
SVD 的一些重要性质和用途包括：
- 信息压缩：由于奇异值是按降序排列的，因此可以通过只保留最大的几个奇异值及其对应的左右奇异向量来近似原始矩阵，从而实现数据的压缩。这种方法在图像压缩、信号处理等领域非常有用。
- 噪声过滤：SVD 可以用于过滤噪声，特别是当噪声主要影响较小的奇异值时。通过去除或减小这些较小的奇异值，可以去除或减轻噪声的影响。
- 主成分分析（PCA）：在统计学中，PCA 可以通过 SVD 来实现。PCA 旨在找到数据中的主成分，这些成分解释了数据中的最大方差。SVD 的左奇异向量对应于 PCA 的主成分方向。
- 求解线性方程组：SVD 可以用于求解线性方程组 Ax=b，特别是当 A 是非方阵或奇异矩阵时。
- 推荐系统：在推荐系统中，SVD 可以用于矩阵补全，即根据部分已知的评分来预测用户对物品的评分。

二、代码实现—SVD奇异值分解方法对图像进行压缩

在 python的NumPy库中，np.linalg.svd() 函数用于计算矩阵的 SVD。
它返回三个对象：左奇异向量 U、奇异值数组 σ（注意这里返回的是一维数组，而不是完整的对角矩阵 Σ）和右奇异向量的转置 V ^T（在 NumPy 中通常直接返回 V，因为 V^T 可以很容易地通过 V.T (转置)获得）。
代码对一张灰度图像进行压缩

代码如下：

import numpy as np
from PIL import Image
import matplotlib.pyplot as plt

""" 定义奇异值分解函数 并返回压缩后的图片和大小"""
# k为压缩的维度
# pic_array 为传入图片的array矩阵格式
def pic_compress(k, pic_array):
    # 定义全局变量
    global u, sigma, vt, sig, new_pic

    u, sigma, vt = np.linalg.svd(pic_array)  # 进行奇异值分解
    sig = np.eye(k) * sigma[: k]     # np.eye 用于生成一个单位对角矩阵
    # np.dot 用于矩阵的乘法运算 (计算：A=UΣV^T ——> 压缩后的图片矩阵)
    new_pic = np.dot(np.dot(u[:, :k], sig), vt[:k, :])  
    size = u.shape[0] * k + sig.shape[0] * sig.shape[1] + k * vt.shape[1]  # 计算压缩后的图片大小
    return new_pic, size

# 读取图片
img = Image.open('lf.jpg')
ori_img = np.array(img)  # 将图片转换成符合 np.linalg.svd() 参数要求的array矩阵格式
# 调用奇异值分解函数
new_img, size = pic_compress(100, ori_img)  # 压缩的维度为100 可设定不同的值查看不同效果

# 分别打印出图片压缩前后的大小
print("original size:" + str(ori_img.shape[0] * ori_img.shape[1]))
print("compress size:" + str(size))

# 创建画布，在同一张画布中显示压缩前后的两张图片
fig, ax = plt.subplots(1, 2)
ax[0].imshow(ori_img, cmap='gray')
ax[0].set_title("before compress")
ax[1].imshow(new_img, cmap='gray')
ax[1].set_title("after compress")
plt.show()