python二维高斯热力图绘制简单的思路代码

zhuyuyaya2024-02-03 21:37

python 复制代码

import numpy as np
import matplotlib.pyplot as plt
from scipy.ndimage import gaussian_filter
import cv2

# 生成一个示例图像
image_size = 100
image = np.zeros((image_size, image_size))

# 在图像中心创建一个高亮区域
center_x, center_y = image_size // 2, image_size // 2
image[center_x - 10:center_x + 10, center_y - 10:center_y + 10] = 1.0

# 对图像进行高斯平滑处理
sigma = 5
smoothed_image = gaussian_filter(image, sigma=sigma)

# 绘制原始图像
plt.subplot(1, 2, 1)
plt.title('Original Image')
plt.imshow(image, cmap='viridis')
plt.colorbar()

# 绘制经过高斯平滑处理后的图像
plt.subplot(1, 2, 2)
plt.title('Smoothed Image (Gaussian Filter)')
plt.imshow(smoothed_image, cmap='viridis')
plt.colorbar()

plt.show()

注意点：
复制代码
image[center_x - 10:center_x + 10, center_y - 10:center_y + 10] = 1.0
里面的1.0表示的是白色

计算机中 0表示黑色,1表示白色,模拟高亮位置

主要变换代码

python 复制代码

import numpy as np
import matplotlib.pyplot as plt
from scipy.ndimage import gaussian_filter
import cv2

# 生成一个示例图像
image_size = 100
image = np.zeros((image_size, image_size))
cv2.imshow("image",image)
cv2.waitKey(0)

# 在图像中心创建一个高亮区域
center_x, center_y = image_size // 2, image_size // 2
image[center_x - 10:center_x + 10, center_y - 10:center_y + 10] = 1.0
cv2.imshow("image",image)
cv2.waitKey(0)
# 高斯平滑处理
sigma = 5
smoothed_image = gaussian_filter(image, sigma=sigma)
cv2.imshow("smoothed_image",smoothed_image)
cv2.waitKey(0)