图片生成全方位图代码

生成全方位图的代码可以分为以下几个步骤：

1. 定义全方位图的尺寸和颜色空间

import cv2
import numpy as np

# 定义全方位图的尺寸和颜色空间
width, height = 640, 480
center_x, center_y = int(width / 2), int(height / 2)
theta = np.linspace(0, 2 * np.pi, width, endpoint=False)
phi = np.linspace(0, np.pi, height)

# 创建全方位图
sphere_image = np.zeros((height, width, 3), dtype=np.uint8)

2. 循环计算每个像素点在球面上的坐标，并转换为图像坐标系中的坐标

for i in range(width):
    for j in range(height):
        x = np.sin(phi[j]) * np.cos(theta[i])
        y = np.sin(phi[j]) * np.sin(theta[i])
        z = np.cos(phi[j])
        # 将球面坐标转换为图像坐标系中的坐标
        u = int(center_x + (width / 2) * x / z)
        v = int(center_y + (height / 2) * y / z)

3. 将每个像素点的颜色设置为球面上对应点的颜色

        # 如果像素点在图像范围内，则设置其颜色为球面上对应点的颜色
        if u >= 0 and u < width and v >= 0 and v < height:
            sphere_image[j, i] = original_image[v, u]

完整代码如下：

import cv2
import numpy as np

# 定义全方位图的尺寸和颜色空间
width, height = 640, 480
center_x, center_y = int(width / 2), int(height / 2)
theta = np.linspace(0, 2 * np.pi, width, endpoint=False)
phi = np.linspace(0, np.pi, height)

# 创建全方位图
sphere_image = np.zeros((height, width, 3), dtype=np.uint8)

# 读取原始图像
original_image = cv2.imread('input.jpg')

# 循环计算每个像素点在球面上的坐标，并转换为图像坐标系中的坐标
for i in range(width):
    for j in range(height):
        x = np.sin(phi[j]) * np.cos(theta[i])
        y = np.sin(phi[j]) * np.sin(theta[i])
        z = np.cos(phi[j])
        # 将球面坐标转换为图像坐标系中的坐标
        u = int(center_x + (width / 2) * x / z)
        v = int(center_y + (height / 2) * y / z)
        # 如果像素点在图像范围内，则设置其颜色为球面上对应点的颜色
        if u >= 0 and u < width and v >= 0 and v < height:
            sphere_image[j, i] = original_image[v, u]

# 保存生成的全方位图
cv2.imwrite('sphere_image.jpg', sphere_image)