cv2+svd分解 压缩图像

import cv2   # v:vision 视觉
import numpy as np
import matplotlib.pyplot as plt

plt.rcParams["font.sans-serif"] = ["SimHei"]

image = cv2.imread("111.jpg")
w = 264
h = 266
# 将原来的图像进行压缩
image = cv2.resize(image, dsize=(h, w))
# 调整RGB三个颜色的顺序-->BGR
image = image[:, :, [2, 1, 0]]  # cv2里用的是BGR不是RGB
# 将BGR颜色的图片变成灰色
image_gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

# svd奇异值分解
u, s, v = np.linalg.svd(image_gray)
le = int(v.shape[0] / 4)
ss = np.diag(s)
u = u[:, :le]
s = ss[:le, :le]
v = v[:le, :]
image2 = u.dot(s).dot(v)

# 将图片进行保存
cv2.imwrite("./0.jpg", image_gray)
cv2.imwrite("./1.jpg", image2)
fig, subplot_arr = plt.subplots(2, 2)
plt.gray()
subplot_arr[0, 0].set_title('原图')
subplot_arr[0, 0].imshow(image)
subplot_arr[0, 1].set_title('灰色图')
subplot_arr[0, 1].imshow(image_gray)
subplot_arr[1, 0].set_title('压缩图')
subplot_arr[1, 0].imshow(image2)
subplot_arr[1, 1].set_title('压缩前后的差值')
subplot_arr[1, 1].imshow(image_gray - image2)
plt.show()