多种插值方法

图像降维，多种插值方法

import cv2 as cv
import numpy as np
from matplotlib import pyplot as plt

# 用来正常显示中文标签
plt.rcParams['font.sans-serif'] = ['SimHei']

# 随机生成一张图片用于测试插值算法
img = np.uint8(np.random.randint(0, 255, size=(1024,1024)))
height,width = img.shape

# 实际输入图片
image_path = './result2-5.bmp'
img_cv = cv.imread(image_path, cv.IMREAD_GRAYSCALE)

image_input = img_cv

# 设定通过插值之后图片的size
new_dimension = (512,512)


def img_draw_subplot(subplot_position,img,title_name,cmap):
    plt.subplot(subplot_position)
    plt.title(title_name)
    plt.imshow(img,cmap)


def image_interpolation(img,new_dimension,inter_method):
    inter_img = cv.resize(img,new_dimension,interpolation=inter_method)
    return inter_img

# 设置cmap
cmap = "gray"
i = 1

name_inter = ["nearest_img2", "linear_img", "cubic_img", "area_img", "lanczos_img"]

# 最近邻插值算法
nearest_img = image_interpolation(image_input,new_dimension,cv.INTER_NEAREST)
cv.imwrite("nearest_img.bmp", nearest_img)
cv.imwrite("nearest_img%d"%(i) + ".bmp", nearest_img)
cv.imwrite(name_inter[0] + ".bmp", nearest_img)
# 双线性插值算法,resize函数默认的插值算法
linear_img = image_interpolation(image_input,new_dimension,cv.INTER_LINEAR)
cv.imwrite("linear_img.bmp", linear_img)
# 三次样条插值算法
cubic_img = image_interpolation(image_input,new_dimension,cv.INTER_CUBIC)
cv.imwrite("cubic_img.bmp", cubic_img)
# 区域插值
area_img = image_interpolation(image_input,new_dimension,cv.INTER_AREA)
cv.imwrite("area_img.bmp", area_img)
# Lanczos插值
lanczos_img = image_interpolation(image_input,new_dimension,cv.INTER_LANCZOS4)
cv.imwrite("lanczos_img.bmp", lanczos_img)

img_draw_subplot(231, image_input, "原图", cmap=cmap)
img_draw_subplot(232, nearest_img, "最近邻插值", cmap=cmap)
img_draw_subplot(233, linear_img, "双线性插值", cmap=cmap)
img_draw_subplot(234, cubic_img, "双三次插值", cmap=cmap)
img_draw_subplot(235, area_img, "区域插值", cmap=cmap)
img_draw_subplot(236, lanczos_img, "Lanczos插值", cmap=cmap)

plt.show()

# 评估图像失真,信噪比