conan/utils/util.py

32 lines
744 B
Python
Raw Permalink Normal View History

2021-10-17 12:49:49 +02:00
import numpy as np
def sperical2equirec(theta, phi, img_w, img_h):
u = np.deg2rad(theta) / (2 * np.pi)
v = np.deg2rad(phi) / np.pi
x = u * img_w
y = v * img_h
return x, y
def equirec2spherical(x, y, img_w, img_h):
# Normalize coordinates
u = x / img_w
v = y / img_h
# Calculate spherical coordinates from normalized coordinates
# theta is horizontal angle, phi is polar angle
theta = u * 2 * np.pi
phi = v * np.pi
return np.rad2deg(theta), np.rad2deg(phi)
def get_circle(radius):
""" helper function returns circle to x, y coordinates"""
theta = np.linspace(0, 2 * np.pi, 100)
x = radius * np.cos(theta)
y = radius * np.sin(theta)
return np.array(x), np.array(y)